vumichien commited on
Commit
074427e
·
1 Parent(s): 89e88d4

add new column to keep

Browse files
Files changed (1) hide show
  1. routes/predict.py +1 -1
routes/predict.py CHANGED
@@ -62,7 +62,7 @@ async def predict(
62
  df_predicted = nameMappingHelper.map_standard_names()
63
 
64
  # Create output dataframe and save to CSV
65
- column_to_keep = ['シート名', '行', '科目', '分類', '名称', '摘要', '備考']
66
  output_df = inputData.dataframe[column_to_keep].copy()
67
  output_df.reset_index(drop=False, inplace=True)
68
  output_df.loc[:, "出力_科目"] = df_predicted["出力_科目"]
 
62
  df_predicted = nameMappingHelper.map_standard_names()
63
 
64
  # Create output dataframe and save to CSV
65
+ column_to_keep = ['シート名', '行', '科目', '中科目', '分類', '名称', '摘要', '備考']
66
  output_df = inputData.dataframe[column_to_keep].copy()
67
  output_df.reset_index(drop=False, inplace=True)
68
  output_df.loc[:, "出力_科目"] = df_predicted["出力_科目"]