|
@@ -60,13 +60,20 @@ class DataProcessor(object):
|
|
|
if label_obj:
|
|
|
total_return = label_obj.get('total_return', 0)
|
|
|
total_view = label_obj.get('total_view', 0)
|
|
|
- if total_view == 0:
|
|
|
- label = None
|
|
|
- else:
|
|
|
- if total_return == 0:
|
|
|
+ if total_return is not None and total_view is not None:
|
|
|
+ total_return = float(total_return)
|
|
|
+ total_view = float(total_view)
|
|
|
+ if total_view == 0:
|
|
|
label = None
|
|
|
else:
|
|
|
- label = float(total_return) / float(total_view)
|
|
|
+ if total_return == 0:
|
|
|
+ label = None
|
|
|
+ else:
|
|
|
+ label = float(total_return) / float(total_view)
|
|
|
+ elif total_return is None and total_view is not None:
|
|
|
+ label = 0
|
|
|
+ else:
|
|
|
+ label = None
|
|
|
else:
|
|
|
label = None
|
|
|
return label, label_dt
|