diff --git a/src/main.py b/src/main.py index c4e9b9a..897a34a 100644 --- a/src/main.py +++ b/src/main.py @@ -72,6 +72,7 @@ def instantiate_model(A, index, pad_index, device): print(cls) return cls, phi + def main(opt): device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu') @@ -236,18 +237,3 @@ if __name__ == '__main__': create_path_if_not_exists(opt.pickle) main(opt) - -""" -python3 main.py -d imdb62 -m savlin -A 10 -e 100 -H 32 -c 64 --lr 0.01 -b 250 -svm: acc=0.915 macrof1=0.915 microf1=0.915 -network prediction (savlin) -acc=90.9000% -macro-f1=0.9086 -micro-f1=0.9090 - -end-to-end network prediction (attr) -acc=94.6000% -macro-f1=0.9458 -micro-f1=0.9460 - -""" \ No newline at end of file