From c40954472c3b4fc33de15f39821e81117ff7d38c Mon Sep 17 00:00:00 2001
From: Siarhei Siniak <siarheisiniak@yahoo.com>
Date: Thu, 15 Jul 2021 09:53:53 +0300
Subject: [PATCH] [~] Refactor

---
 python/tasks/jigsaw_toxic.py | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/python/tasks/jigsaw_toxic.py b/python/tasks/jigsaw_toxic.py
index bf1f6b2..56298c8 100644
--- a/python/tasks/jigsaw_toxic.py
+++ b/python/tasks/jigsaw_toxic.py
@@ -247,12 +247,16 @@ def kernel_3(
         return roc_auc
 
     # %% [code]
-    o_2['model'].fit(
-        o_2['xtrain_pad'],
-        o_2['ytrain'],
-        nb_epoch=5,
-        batch_size=64*o_2['strategy'].num_replicas_in_sync
-    ) #Multiplying by Strategy to run on TPU's
+    if os.path.exists('model.h5'):
+        o_2['model'].load_weights('model.h5')
+    else:
+        o_2['model'].fit(
+            o_2['xtrain_pad'],
+            o_2['ytrain'],
+            nb_epoch=5,
+            batch_size=64*o_2['strategy'].num_replicas_in_sync
+        ) #Multiplying by Strategy to run on TPU's
+        o_2['model'].save_weights('model.h5')
 
     # %% [code]
     scores = o_2['model'].predict(o_2['xvalid_pad'])