ReadMe

ashfarhangi · May 21, 2022 · 1e722a8 · 1e722a8
1 parent 9d2a4fe
commit 1e722a8
Show file tree

Hide file tree

Showing 6 changed files with 4 additions and 1 deletion.
diff --git a/data/__pycache__/dataloader.cpython-38.pyc b/data/__pycache__/dataloader.cpython-38.pyc
diff --git a/data/dataloader.py b/data/dataloader.py
@@ -14,6 +14,7 @@
 import warnings
 warnings.filterwarnings('ignore')
 import sklearn
+from sklearn.preprocessing import LabelEncoder
 from sklearn.model_selection import train_test_split
 from sklearn.metrics import confusion_matrix,classification_report
 from collections import defaultdict

diff --git a/main.py b/main.py
@@ -4,6 +4,8 @@
 from src import utils
 from data import dataloader
 from src import metric,model,plot,trainer
+from src.model import Protoformer 
+
 
 
 def run():
@@ -17,4 +19,4 @@ def run():
     model.evaluate()
 
 if __name__ == '__main__':
-    run()    
+    run()
diff --git a/src/__pycache__/metric.cpython-38.pyc b/src/__pycache__/metric.cpython-38.pyc
diff --git a/src/__pycache__/model.cpython-38.pyc b/src/__pycache__/model.cpython-38.pyc
diff --git a/src/__pycache__/utils.cpython-38.pyc b/src/__pycache__/utils.cpython-38.pyc