Hello World RecBox!

- add main.py
RUCAIBox · Jun 24, 2020 · a0d6d7b · a0d6d7b
1 parent 5a42d68
commit a0d6d7b
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/main.py b/main.py
@@ -0,0 +1,24 @@
+from config import Config
+from dataset import Dataset
+from model import Model
+from trainer import Trainer
+from evaluator import Evaluator
+
+
+
+config = Config()
+config.init()
+
+
+dataset = Dataset(config)
+train_data, test_data = dataset.preprocessing()
+
+
+model = Model(config)
+trainer = Trainer(config)
+model = trainer.train(model, train_data)
+result = trainer.predict(model, test_data)
+
+
+evaluator = Evaluator(config)
+evaluator.evaluate(result, test_data)