train model

Jenkins/Jenkinsfile.create_dataset

@@ -34,7 +34,7 @@ node {
             }
         }
         stage('Archive artifacts') {
-            archiveArtifacts "dataset.csv, train_data.csv, test_data.csv, dev_data.csv, neural_network_evaluation.txt"
+            archiveArtifacts "dataset.csv, train_data.csv, test_data.csv, dev_data.csv"
         }
     }
 }

Jenkins/Jenkinsfile.training

@@ -23,5 +23,8 @@ node {
                 sh "python3 Scripts/train_neural_network.py $TRAIN_ARGS"
             }
         }
+        stage('Archive artifacts') {
+            archiveArtifacts "neural_network_evaluation.txt, neural_network_model/**/*"
+        }
     }
 }

Scripts/train_neural_network.py

@@ -82,7 +82,7 @@ def main():
     x_train, x_test, vocab_size = tokenize(pd.concat([x_train, x_test]), x_train, x_test, 100)
     model = get_model(50, vocab_size)
     train_model(model, x_train, y_train)
-    # save_model(model, abs_data_path, 'neural_network')
+    save_model(model, abs_data_path, 'neural_network_model')
     evaluate_and_save(model, x_test, y_test, abs_data_path)