(args)
| 188 | return preds |
| 189 | |
| 190 | def main(args): |
| 191 | subjects = sorted([f.split(".xlsx")[0] for f in os.listdir(os.path.join(args.data_dir, "test")) if ".xlsx" in f]) |
| 192 | if not os.path.exists(args.save_dir): |
| 193 | os.mkdir(args.save_dir) |
| 194 | |
| 195 | print("subjects:", subjects) |
| 196 | print("args", args) |
| 197 | |
| 198 | for subject in subjects: |
| 199 | if subject != "医疗": |
| 200 | dev_df = pd.read_excel(os.path.join(args.data_dir, "dev", subject + ".xlsx"), header=0)[:args.ntrain] |
| 201 | test_df = pd.read_excel(os.path.join(args.data_dir, "test", subject + ".xlsx"), header=0) |
| 202 | preds = eval(args, subject, dev_df, test_df) |
| 203 | writelines_to_file(os.path.join(args.save_dir, subject), preds) |
| 204 | |
| 205 | else: |
| 206 | dev_df = pd.read_excel(os.path.join(args.data_dir, "dev", subject + ".xlsx"), header=0)[:args.ntrain] |
| 207 | f = pd.ExcelFile(os.path.join(args.data_dir, "test", subject + ".xlsx")) |
| 208 | sheet_list = f.sheet_names |
| 209 | for sheet in sheet_list: |
| 210 | test_df = pd.read_excel(os.path.join(args.data_dir, "test", subject + ".xlsx"), header=0, sheet_name=sheet) |
| 211 | preds = eval(args, sheet, dev_df, test_df) |
| 212 | writelines_to_file(os.path.join(args.save_dir, sheet), preds) |
| 213 | |
| 214 | if __name__ == "__main__": |
| 215 | parser = argparse.ArgumentParser() |
no test coverage detected