From f5f40c82695bb46801190518f26e1d326996b69a 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:36 +0200
Subject: [PATCH] Ajout de la tache train_model + update .gitignore

---
 .gitignore | 5 +++++
 Makefile   | 9 +++++++++
 2 files changed, 14 insertions(+)

diff --git a/.gitignore b/.gitignore
index ad7ed72..afde035 100644
--- a/.gitignore
+++ b/.gitignore
@@ -80,3 +80,8 @@ target/
 
 # Mac OS-specific storage files
 .DS_Store
+
+# makefile step
+sync_collections
+resize_collections
+build_tfrecords
\ No newline at end of file
diff --git a/Makefile b/Makefile
index 990ad3d..2d9fb48 100644
--- a/Makefile
+++ b/Makefile
@@ -39,10 +39,19 @@ resize_collections: src/data/resize_collections.py
 	$(PYTHON_INTERPRETER) src/data/resize_collections.py
 	touch resize_collections
 
+## Build TFRecords
 build_tfrecords: sync_collections resize_collections src/features/build_tfrecords.py
 	$(PYTHON_INTERPRETER) src/features/build_tfrecords.py
 	touch build_tfrecords
 
+## Train model 
+train_model: build_tfrecords
+	$(PYTHON_INTERPRETER) src/models/train_model.py
+
+## test
+draft: 
+	$(PYTHON_INTERPRETER) src/models/data_loader.py
+
 
 ## Delete all compiled Python files
 clean: