def csv_to_array(csv_filedir):
"""
save array to scv
"""
csv_table = pd.read_csv(csv_filedir)
sorted_table = csv_table.sort_values(by='VideoId')
sorted_table = sorted_table.reset_index(drop=True)
label_array = np.zeros((len(sorted_table), 4716), dtype=np.float16)
label_pair = sorted_table['LabelConfidencePairs']
videoid = sorted_table['VideoId']
for i in range(len(label_pair)):
example = label_pair[i].split(' ')
for j in range(0, len(example), 2):
label_array[i, int(example[j])] = float(example[j + 1])
return label_array, videoid
评论列表
文章目录