|
|
@@ -70,6 +70,8 @@ def process_predict_data(filename):
|
|
|
filtered_videos = [str(item) for item in filter_video_status(video_ids=video_id_list)]
|
|
|
data = data.loc[data['videoid'].isin(filtered_videos)]
|
|
|
|
|
|
+ video_id_final = data['videoid']
|
|
|
+
|
|
|
# 获取x
|
|
|
drop_columns = ['videoid', 'dt', 'futre7dayreturn', 'videotags', 'words_without_tags']
|
|
|
x = data.drop(columns=drop_columns)
|
|
|
@@ -85,7 +87,7 @@ def process_predict_data(filename):
|
|
|
# 缺失值填充为0
|
|
|
x.fillna(0)
|
|
|
|
|
|
- return x, filtered_videos
|
|
|
+ return x, video_id_final
|
|
|
|
|
|
|
|
|
def train(x, y, features):
|
