Merge pull request #74 from bcaitech1/LGBM

[창우] add LGBM
bcaitech1 · Jul 29, 2021 · bd0f08a · bd0f08a
2 parents 02c3ddc + 85a8673
commit bd0f08a
Show file tree

Hide file tree

Showing 14 changed files with 153 additions and 1,433 deletions.
diff --git a/.gitignore b/.gitignore
@@ -18,3 +18,6 @@ __pycache__
 
 # ignore .idea folder
 .idea
+
+# pycaret log
+*.log
diff --git a/No Caffeine No Gain.pdf b/No Caffeine No Gain.pdf
diff --git a/args.py b/args.py
@@ -71,7 +71,11 @@ def parse_args(mode='train'):
     # Pseudo Labeling
     parser.add_argument('--use_pseudo', default=False, type=bool, help='Using Pseudo labeling')
     parser.add_argument('--pseudo_label_file', default='', type=str, help='file path for pseudo labeling')
-
+
+    # Finetuning
+    parser.add_argument('--use_finetune', default=False, type=bool, help='Using Fine Tuning')
+    parser.add_argument('--trained_model', default='/opt/ml/code/p4-dkt-no_caffeine_no_gain/models/re_pse_Bert_40_5/model_epoch7.pt', type=str, help='pretrained model path')
+
     # log
     parser.add_argument('--log_steps', default=50, type=int, help='print log per n steps')