wip

imports, add tensorboard logs, pipenv
cli args, gitignore

Author: Mathis Chenuet

.gitignore

@@ -102,3 +102,6 @@ venv.bak/
 
 # mypy
 .mypy_cache/
+
+# tensorboard event files
+*out.tfevents*

Pipfile

@@ -6,6 +6,8 @@ name = "pypi"
 [packages]
 pytorch-pretrained-bert = "*"
 attrs = "*"
+tensorboardx = "*"
+sklearn = "*"
 
 [dev-packages]
 

Pipfile.lock

@@ -6,9 +6,13 @@
     parser = ArgumentParser(formatter_class=ArgumentDefaultsHelpFormatter)
     arg = parser.add_argument
 
-    arg('--data_dir', default='data/sentihood/')
+    arg('--data_dir', default='data/sentihood/', help="Path to the dataset directory")
+    arg('-b', '--batch_size', default=4, help="Batch size for training and evaluation")
+    arg('-lr', '--learning_rate', default=1e-4, help="Optimizer learning rate")
+    arg('-wd', '--weight_decay', default=0.01, help="Optimizer weight decay (L2)")
 
     args = parser.parse_args()
 
     from sentihood import main
-    main(**vars(args))
+    # main(**vars(args))
+    main(args)