work on lab6

lab6/Jenkinsfile

@@ -11,8 +11,9 @@ pipeline {
 					echo 'Datasets copied'
 					echo 'Conducting simple regression model test'
 					sh 'python3 simple_regression.py'
-					echo 'Model predictions saved'
-                    sh 'head lego_linreg_results.csv'
+					echo 'Model and predictions saved'
+                    sh 'head lego_reg_results.csv'
+					sh 'head lego_reg_model'
 				}
             }
         }		

simple_regression.py

@@ -1,5 +1,6 @@
 import tensorflow as tf
 from keras import layers
+from keras import save_model
 import pandas as pd
 import numpy as np
 import matplotlib.pyplot as plt
@@ -51,7 +52,10 @@ y = model.predict(x)
 
 # Zapis predykcji do pliku
 results = pd.DataFrame({"input_piece_count": x.numpy().tolist(), "predicted_price": [a[0] for a in y.tolist()]})
-results.to_csv(r'lego_linreg_results.csv', index=False, header=True)
+results.to_csv(r'lego_reg_results.csv', index=False, header=True)
+
+# Zapis modelu do pliku
+model.save('lego_reg_model')
 
 # Opcjonalne statystyki, wykresy
 '''