Esempio n. 1
0
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=',')
Esempio n. 3
0
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')
Esempio n. 4
0
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=',')