(args)
| 162 | return preds |
| 163 | |
| 164 | def main(args): |
| 165 | subjects = sorted([f.split(".xlsx")[0] for f in os.listdir(os.path.join(args.data_dir, "test")) if ".xlsx" in f]) |
| 166 | if not os.path.exists(args.save_dir): |
| 167 | os.mkdir(args.save_dir) |
| 168 | |
| 169 | print("subjects:", subjects) |
| 170 | print("args", args) |
| 171 | |
| 172 | for subject in subjects: |
| 173 | if subject != "医疗": |
| 174 | dev_df = pd.read_excel(os.path.join(args.data_dir, "dev", subject + ".xlsx"), header=0)[:args.ntrain] |
| 175 | test_df = pd.read_excel(os.path.join(args.data_dir, "test", subject + ".xlsx"), header=0) |
| 176 | preds = eval(args, subject, dev_df, test_df) |
| 177 | writelines_to_file(os.path.join(args.save_dir, subject), preds) |
| 178 | |
| 179 | else: |
| 180 | dev_df = pd.read_excel(os.path.join(args.data_dir, "dev", subject + ".xlsx"), header=0)[:args.ntrain] |
| 181 | f = pd.ExcelFile(os.path.join(args.data_dir, "test", subject + ".xlsx")) |
| 182 | sheet_list = f.sheet_names |
| 183 | for sheet in sheet_list: |
| 184 | test_df = pd.read_excel(os.path.join(args.data_dir, "test", subject + ".xlsx"), header=0, sheet_name=sheet) |
| 185 | preds = eval(args, sheet, dev_df, test_df) |
| 186 | writelines_to_file(os.path.join(args.save_dir, sheet), preds) |
| 187 | |
| 188 | if __name__ == "__main__": |
| 189 | parser = argparse.ArgumentParser() |
no test coverage detected