|
|
@@ -36,7 +36,7 @@ if __name__ == '__main__':
|
|
|
# 2. 获取视频特征数据
|
|
|
video_filepath = f"{predict_data_dir}/{video_filename}"
|
|
|
video_df = read_csv_data(filepath=video_filepath)
|
|
|
- video_df = video_df[video_df['videoid' != '-1']]
|
|
|
+ video_df = video_df[video_df['videoid'] != '-1']
|
|
|
print(f"video_df shape: {video_df.shape}")
|
|
|
# 3. 用户特征和视频特征进行拼接
|
|
|
video_features = [
|
