def save_predict_result(results, params): # 读取结果 test_df = pd.read_csv(params['test_x_dir']) # 填充结果 test_df['Prediction'] = results[:20000] # 提取ID和预测结果两列 test_df = test_df[['QID', 'Prediction']] # 保存结果. result_save_path = get_result_filename(params) test_df.to_csv(result_save_path, index=None, sep=',')
def save_predict_result(results, params): sample_size = params['sample_size'] # 读取结果 AutoMaster_TestSet.csv test_df = pd.read_csv(params['test_x_dir']) # 填充结果 test_df['Prediction'] = results[:sample_size] # 提取ID和预测结果两列 test_df = test_df[['QID', 'Prediction']] # 保存结果. result_save_path = get_result_filename(params) test_df.to_csv(result_save_path, index=None, sep=',')
def save_predict_result(results, params): # 读取结果 test_df = pd.read_csv(params['test_x_dir']) # 填充结果 print(len(results)) test_df['Prediction'] = results[:params['num_to_test']] + [''] * ( 20000 - params['num_to_test']) # 提取ID和预测结果两列 test_df = test_df[['QID', 'Prediction']] # 保存结果. result_save_path = get_result_filename(params) test_df.to_csv(result_save_path, index=None, sep=',', encoding='utf8')
def save_predict_result(results, params): # 读取结果 test_df = pd.read_csv(params['test_x_dir']) # 填充结果 print("result",len(results)) #防止写文件失败导致结果丢失,将results也先保存下来 r_save=pd.DataFrame(columns=['result'],data=results) r_save.to_csv("result_save.csv", index=None, sep=',') test_df['Prediction'] = results[:TEST_NUM_SAMPLES]#200000 # 提取ID和预测结果两列 test_df = test_df[['QID', 'Prediction']] # 保存结果. result_save_path = get_result_filename(params) test_df.to_csv(result_save_path, index=None, sep=',')