From f731a2cc6a64a8cbf935296224f1c084290ff4cb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=C3=A7ois=20Vieille?= <vieille.francois@gmail.com>
Date: Thu, 23 Aug 2018 18:56:10 +0200
Subject: [PATCH] script pour lancer le train du modele horizontal/vertical

---
 src/models/train_model.py | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/src/models/train_model.py b/src/models/train_model.py
index e69de29..1e0e919 100644
--- a/src/models/train_model.py
+++ b/src/models/train_model.py
@@ -0,0 +1,22 @@
+import tensorflow as tf
+import os
+import sys
+from dotenv import find_dotenv, load_dotenv
+from data_loader import TFRecordsLoader
+from trainer_model import BaseTrainer
+from base_model import BaseModel
+
+
+def main():
+	load_dotenv(find_dotenv())
+
+	sess = tf.Session()
+	
+	model = BaseModel()
+	data_loader = TFRecordsLoader()
+
+	trainer = BaseTrainer(sess, model, data_loader)
+	trainer.train()
+
+if __name__ == '__main__':
+	main()