diff --git a/Jenkinsfile b/Jenkinsfile
index 92cd3c6..14d8f45 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -32,6 +32,8 @@ pipeline {
                 steps
                 {
                        sh 'python3 train.py ${BATCH_SIZE} ${EPOCHS}'
+                       sh 'rm -r my_model'
+                       sh 'python3 vgsales-mlflow.py ${BATCH_SIZE} ${EPOCHS}'
                 }
             }
 
@@ -40,6 +42,7 @@ pipeline {
 			steps{
 				archiveArtifacts 'vgsales_model.h5'
             archiveArtifacts 'my_runs/**'
+            archiveArtifacts 'my_model/**'
 
 			}
 		}
diff --git a/mlruns/0/13e3e1a2ce674a07b82227babfec5f00/meta.yaml b/mlruns/0/13e3e1a2ce674a07b82227babfec5f00/meta.yaml
new file mode 100644
index 0000000..4103326
--- /dev/null
+++ b/mlruns/0/13e3e1a2ce674a07b82227babfec5f00/meta.yaml
@@ -0,0 +1,15 @@
+artifact_uri: file:///home/tomasz/IUM/ium_434695/mlruns/0/13e3e1a2ce674a07b82227babfec5f00/artifacts
+end_time: null
+entry_point_name: ''
+experiment_id: '0'
+lifecycle_stage: active
+name: ''
+run_id: 13e3e1a2ce674a07b82227babfec5f00
+run_uuid: 13e3e1a2ce674a07b82227babfec5f00
+source_name: ''
+source_type: 4
+source_version: ''
+start_time: 1621790521589
+status: 1
+tags: []
+user_id: tomasz
diff --git a/mlruns/0/13e3e1a2ce674a07b82227babfec5f00/tags/mlflow.source.git.commit b/mlruns/0/13e3e1a2ce674a07b82227babfec5f00/tags/mlflow.source.git.commit
new file mode 100644
index 0000000..2e3fdf7
--- /dev/null
+++ b/mlruns/0/13e3e1a2ce674a07b82227babfec5f00/tags/mlflow.source.git.commit
@@ -0,0 +1 @@
+db58b281061f170a706135380f1e7c86c7162acc
\ No newline at end of file
diff --git a/mlruns/0/13e3e1a2ce674a07b82227babfec5f00/tags/mlflow.source.name b/mlruns/0/13e3e1a2ce674a07b82227babfec5f00/tags/mlflow.source.name
new file mode 100644
index 0000000..939bfd8
--- /dev/null
+++ b/mlruns/0/13e3e1a2ce674a07b82227babfec5f00/tags/mlflow.source.name
@@ -0,0 +1 @@
+/home/tomasz/IUM/ium_434695/vgsales-mlflow.py
\ No newline at end of file
diff --git a/mlruns/0/13e3e1a2ce674a07b82227babfec5f00/tags/mlflow.source.type b/mlruns/0/13e3e1a2ce674a07b82227babfec5f00/tags/mlflow.source.type
new file mode 100644
index 0000000..0c2c1fe
--- /dev/null
+++ b/mlruns/0/13e3e1a2ce674a07b82227babfec5f00/tags/mlflow.source.type
@@ -0,0 +1 @@
+LOCAL
\ No newline at end of file
diff --git a/mlruns/0/13e3e1a2ce674a07b82227babfec5f00/tags/mlflow.user b/mlruns/0/13e3e1a2ce674a07b82227babfec5f00/tags/mlflow.user
new file mode 100644
index 0000000..4b82f83
--- /dev/null
+++ b/mlruns/0/13e3e1a2ce674a07b82227babfec5f00/tags/mlflow.user
@@ -0,0 +1 @@
+tomasz
\ No newline at end of file
diff --git a/mlruns/0/1797ad0fc8144ad48230f8965a689293/meta.yaml b/mlruns/0/1797ad0fc8144ad48230f8965a689293/meta.yaml
new file mode 100644
index 0000000..ca17e6e
--- /dev/null
+++ b/mlruns/0/1797ad0fc8144ad48230f8965a689293/meta.yaml
@@ -0,0 +1,15 @@
+artifact_uri: file:///home/tomasz/IUM/ium_434695/mlruns/0/1797ad0fc8144ad48230f8965a689293/artifacts
+end_time: 1621790489715
+entry_point_name: ''
+experiment_id: '0'
+lifecycle_stage: active
+name: ''
+run_id: 1797ad0fc8144ad48230f8965a689293
+run_uuid: 1797ad0fc8144ad48230f8965a689293
+source_name: ''
+source_type: 4
+source_version: ''
+start_time: 1621790450170
+status: 4
+tags: []
+user_id: tomasz
diff --git a/mlruns/0/1797ad0fc8144ad48230f8965a689293/metrics/rmse b/mlruns/0/1797ad0fc8144ad48230f8965a689293/metrics/rmse
new file mode 100644
index 0000000..57f3bd1
--- /dev/null
+++ b/mlruns/0/1797ad0fc8144ad48230f8965a689293/metrics/rmse
@@ -0,0 +1 @@
+1621790489709 0.032775074938251283 0
diff --git a/mlruns/0/1797ad0fc8144ad48230f8965a689293/params/batch_size b/mlruns/0/1797ad0fc8144ad48230f8965a689293/params/batch_size
new file mode 100644
index 0000000..19c7bdb
--- /dev/null
+++ b/mlruns/0/1797ad0fc8144ad48230f8965a689293/params/batch_size
@@ -0,0 +1 @@
+16
\ No newline at end of file
diff --git a/mlruns/0/1797ad0fc8144ad48230f8965a689293/params/epochs b/mlruns/0/1797ad0fc8144ad48230f8965a689293/params/epochs
new file mode 100644
index 0000000..3f10ffe
--- /dev/null
+++ b/mlruns/0/1797ad0fc8144ad48230f8965a689293/params/epochs
@@ -0,0 +1 @@
+15
\ No newline at end of file
diff --git a/mlruns/0/1797ad0fc8144ad48230f8965a689293/tags/mlflow.source.git.commit b/mlruns/0/1797ad0fc8144ad48230f8965a689293/tags/mlflow.source.git.commit
new file mode 100644
index 0000000..2e3fdf7
--- /dev/null
+++ b/mlruns/0/1797ad0fc8144ad48230f8965a689293/tags/mlflow.source.git.commit
@@ -0,0 +1 @@
+db58b281061f170a706135380f1e7c86c7162acc
\ No newline at end of file
diff --git a/mlruns/0/1797ad0fc8144ad48230f8965a689293/tags/mlflow.source.name b/mlruns/0/1797ad0fc8144ad48230f8965a689293/tags/mlflow.source.name
new file mode 100644
index 0000000..939bfd8
--- /dev/null
+++ b/mlruns/0/1797ad0fc8144ad48230f8965a689293/tags/mlflow.source.name
@@ -0,0 +1 @@
+/home/tomasz/IUM/ium_434695/vgsales-mlflow.py
\ No newline at end of file
diff --git a/mlruns/0/1797ad0fc8144ad48230f8965a689293/tags/mlflow.source.type b/mlruns/0/1797ad0fc8144ad48230f8965a689293/tags/mlflow.source.type
new file mode 100644
index 0000000..0c2c1fe
--- /dev/null
+++ b/mlruns/0/1797ad0fc8144ad48230f8965a689293/tags/mlflow.source.type
@@ -0,0 +1 @@
+LOCAL
\ No newline at end of file
diff --git a/mlruns/0/1797ad0fc8144ad48230f8965a689293/tags/mlflow.user b/mlruns/0/1797ad0fc8144ad48230f8965a689293/tags/mlflow.user
new file mode 100644
index 0000000..4b82f83
--- /dev/null
+++ b/mlruns/0/1797ad0fc8144ad48230f8965a689293/tags/mlflow.user
@@ -0,0 +1 @@
+tomasz
\ No newline at end of file
diff --git a/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/artifacts/vgsales_model.h5/MLmodel b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/artifacts/vgsales_model.h5/MLmodel
new file mode 100644
index 0000000..c8e18b2
--- /dev/null
+++ b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/artifacts/vgsales_model.h5/MLmodel
@@ -0,0 +1,14 @@
+artifact_path: vgsales_model.h5
+flavors:
+  keras:
+    data: data
+    keras_module: tensorflow.keras
+    keras_version: 2.4.0
+    save_format: tf
+  python_function:
+    data: data
+    env: conda.yaml
+    loader_module: mlflow.keras
+    python_version: 3.8.5
+run_id: 1d5601fad0dc496b8e16c284d2fcd1c9
+utc_time_created: '2021-05-23 17:16:38.081474'
diff --git a/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/artifacts/vgsales_model.h5/conda.yaml b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/artifacts/vgsales_model.h5/conda.yaml
new file mode 100644
index 0000000..3fa4018
--- /dev/null
+++ b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/artifacts/vgsales_model.h5/conda.yaml
@@ -0,0 +1,10 @@
+channels:
+- defaults
+- conda-forge
+dependencies:
+- python=3.8.5
+- pip
+- pip:
+  - mlflow
+  - tensorflow==2.4.1
+name: mlflow-env
diff --git a/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/artifacts/vgsales_model.h5/data/keras_module.txt b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/artifacts/vgsales_model.h5/data/keras_module.txt
new file mode 100644
index 0000000..2c73dfd
--- /dev/null
+++ b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/artifacts/vgsales_model.h5/data/keras_module.txt
@@ -0,0 +1 @@
+tensorflow.keras
\ No newline at end of file
diff --git a/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/artifacts/vgsales_model.h5/data/model/saved_model.pb b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/artifacts/vgsales_model.h5/data/model/saved_model.pb
new file mode 100644
index 0000000..2c3fec8
Binary files /dev/null and b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/artifacts/vgsales_model.h5/data/model/saved_model.pb differ
diff --git a/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/artifacts/vgsales_model.h5/data/model/variables/variables.data-00000-of-00001 b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/artifacts/vgsales_model.h5/data/model/variables/variables.data-00000-of-00001
new file mode 100644
index 0000000..d220ab5
Binary files /dev/null and b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/artifacts/vgsales_model.h5/data/model/variables/variables.data-00000-of-00001 differ
diff --git a/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/artifacts/vgsales_model.h5/data/model/variables/variables.index b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/artifacts/vgsales_model.h5/data/model/variables/variables.index
new file mode 100644
index 0000000..33c6728
Binary files /dev/null and b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/artifacts/vgsales_model.h5/data/model/variables/variables.index differ
diff --git a/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/artifacts/vgsales_model.h5/data/save_format.txt b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/artifacts/vgsales_model.h5/data/save_format.txt
new file mode 100644
index 0000000..439ab29
--- /dev/null
+++ b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/artifacts/vgsales_model.h5/data/save_format.txt
@@ -0,0 +1 @@
+tf
\ No newline at end of file
diff --git a/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/meta.yaml b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/meta.yaml
new file mode 100644
index 0000000..764916f
--- /dev/null
+++ b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/meta.yaml
@@ -0,0 +1,15 @@
+artifact_uri: file:///home/tomasz/IUM/ium_434695/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/artifacts
+end_time: 1621790199119
+entry_point_name: ''
+experiment_id: '0'
+lifecycle_stage: active
+name: ''
+run_id: 1d5601fad0dc496b8e16c284d2fcd1c9
+run_uuid: 1d5601fad0dc496b8e16c284d2fcd1c9
+source_name: ''
+source_type: 4
+source_version: ''
+start_time: 1621790167199
+status: 4
+tags: []
+user_id: tomasz
diff --git a/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/metrics/rmse b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/metrics/rmse
new file mode 100644
index 0000000..4b2e081
--- /dev/null
+++ b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/metrics/rmse
@@ -0,0 +1 @@
+1621790198079 0.031762412384138385 0
diff --git a/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/params/batch_size b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/params/batch_size
new file mode 100644
index 0000000..19c7bdb
--- /dev/null
+++ b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/params/batch_size
@@ -0,0 +1 @@
+16
\ No newline at end of file
diff --git a/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/params/epochs b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/params/epochs
new file mode 100644
index 0000000..3f10ffe
--- /dev/null
+++ b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/params/epochs
@@ -0,0 +1 @@
+15
\ No newline at end of file
diff --git a/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/tags/mlflow.log-model.history b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/tags/mlflow.log-model.history
new file mode 100644
index 0000000..0e77da3
--- /dev/null
+++ b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/tags/mlflow.log-model.history
@@ -0,0 +1 @@
+[{"run_id": "1d5601fad0dc496b8e16c284d2fcd1c9", "artifact_path": "vgsales_model.h5", "utc_time_created": "2021-05-23 17:16:38.081474", "flavors": {"keras": {"keras_module": "tensorflow.keras", "keras_version": "2.4.0", "save_format": "tf", "data": "data"}, "python_function": {"loader_module": "mlflow.keras", "python_version": "3.8.5", "data": "data", "env": "conda.yaml"}}}]
\ No newline at end of file
diff --git a/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/tags/mlflow.source.git.commit b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/tags/mlflow.source.git.commit
new file mode 100644
index 0000000..2e3fdf7
--- /dev/null
+++ b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/tags/mlflow.source.git.commit
@@ -0,0 +1 @@
+db58b281061f170a706135380f1e7c86c7162acc
\ No newline at end of file
diff --git a/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/tags/mlflow.source.name b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/tags/mlflow.source.name
new file mode 100644
index 0000000..939bfd8
--- /dev/null
+++ b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/tags/mlflow.source.name
@@ -0,0 +1 @@
+/home/tomasz/IUM/ium_434695/vgsales-mlflow.py
\ No newline at end of file
diff --git a/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/tags/mlflow.source.type b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/tags/mlflow.source.type
new file mode 100644
index 0000000..0c2c1fe
--- /dev/null
+++ b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/tags/mlflow.source.type
@@ -0,0 +1 @@
+LOCAL
\ No newline at end of file
diff --git a/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/tags/mlflow.user b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/tags/mlflow.user
new file mode 100644
index 0000000..4b82f83
--- /dev/null
+++ b/mlruns/0/1d5601fad0dc496b8e16c284d2fcd1c9/tags/mlflow.user
@@ -0,0 +1 @@
+tomasz
\ No newline at end of file
diff --git a/mlruns/0/5d3630baec6a424da8572f7ef9ef85b3/meta.yaml b/mlruns/0/5d3630baec6a424da8572f7ef9ef85b3/meta.yaml
new file mode 100644
index 0000000..b0ab1f4
--- /dev/null
+++ b/mlruns/0/5d3630baec6a424da8572f7ef9ef85b3/meta.yaml
@@ -0,0 +1,15 @@
+artifact_uri: file:///home/tomasz/IUM/ium_434695/mlruns/0/5d3630baec6a424da8572f7ef9ef85b3/artifacts
+end_time: 1621790904083
+entry_point_name: ''
+experiment_id: '0'
+lifecycle_stage: active
+name: ''
+run_id: 5d3630baec6a424da8572f7ef9ef85b3
+run_uuid: 5d3630baec6a424da8572f7ef9ef85b3
+source_name: ''
+source_type: 4
+source_version: ''
+start_time: 1621790867133
+status: 4
+tags: []
+user_id: tomasz
diff --git a/mlruns/0/5d3630baec6a424da8572f7ef9ef85b3/metrics/rmse b/mlruns/0/5d3630baec6a424da8572f7ef9ef85b3/metrics/rmse
new file mode 100644
index 0000000..c1fe1d6
--- /dev/null
+++ b/mlruns/0/5d3630baec6a424da8572f7ef9ef85b3/metrics/rmse
@@ -0,0 +1 @@
+1621790904078 0.03182406217035666 0
diff --git a/mlruns/0/5d3630baec6a424da8572f7ef9ef85b3/params/batch_size b/mlruns/0/5d3630baec6a424da8572f7ef9ef85b3/params/batch_size
new file mode 100644
index 0000000..19c7bdb
--- /dev/null
+++ b/mlruns/0/5d3630baec6a424da8572f7ef9ef85b3/params/batch_size
@@ -0,0 +1 @@
+16
\ No newline at end of file
diff --git a/mlruns/0/5d3630baec6a424da8572f7ef9ef85b3/params/epochs b/mlruns/0/5d3630baec6a424da8572f7ef9ef85b3/params/epochs
new file mode 100644
index 0000000..3f10ffe
--- /dev/null
+++ b/mlruns/0/5d3630baec6a424da8572f7ef9ef85b3/params/epochs
@@ -0,0 +1 @@
+15
\ No newline at end of file
diff --git a/mlruns/0/5d3630baec6a424da8572f7ef9ef85b3/tags/mlflow.source.git.commit b/mlruns/0/5d3630baec6a424da8572f7ef9ef85b3/tags/mlflow.source.git.commit
new file mode 100644
index 0000000..2e3fdf7
--- /dev/null
+++ b/mlruns/0/5d3630baec6a424da8572f7ef9ef85b3/tags/mlflow.source.git.commit
@@ -0,0 +1 @@
+db58b281061f170a706135380f1e7c86c7162acc
\ No newline at end of file
diff --git a/mlruns/0/5d3630baec6a424da8572f7ef9ef85b3/tags/mlflow.source.name b/mlruns/0/5d3630baec6a424da8572f7ef9ef85b3/tags/mlflow.source.name
new file mode 100644
index 0000000..939bfd8
--- /dev/null
+++ b/mlruns/0/5d3630baec6a424da8572f7ef9ef85b3/tags/mlflow.source.name
@@ -0,0 +1 @@
+/home/tomasz/IUM/ium_434695/vgsales-mlflow.py
\ No newline at end of file
diff --git a/mlruns/0/5d3630baec6a424da8572f7ef9ef85b3/tags/mlflow.source.type b/mlruns/0/5d3630baec6a424da8572f7ef9ef85b3/tags/mlflow.source.type
new file mode 100644
index 0000000..0c2c1fe
--- /dev/null
+++ b/mlruns/0/5d3630baec6a424da8572f7ef9ef85b3/tags/mlflow.source.type
@@ -0,0 +1 @@
+LOCAL
\ No newline at end of file
diff --git a/mlruns/0/5d3630baec6a424da8572f7ef9ef85b3/tags/mlflow.user b/mlruns/0/5d3630baec6a424da8572f7ef9ef85b3/tags/mlflow.user
new file mode 100644
index 0000000..4b82f83
--- /dev/null
+++ b/mlruns/0/5d3630baec6a424da8572f7ef9ef85b3/tags/mlflow.user
@@ -0,0 +1 @@
+tomasz
\ No newline at end of file
diff --git a/mlruns/0/794e3e4e628e416289e6237f26ef142d/meta.yaml b/mlruns/0/794e3e4e628e416289e6237f26ef142d/meta.yaml
new file mode 100644
index 0000000..5eca9c9
--- /dev/null
+++ b/mlruns/0/794e3e4e628e416289e6237f26ef142d/meta.yaml
@@ -0,0 +1,15 @@
+artifact_uri: file:///home/tomasz/IUM/ium_434695/mlruns/0/794e3e4e628e416289e6237f26ef142d/artifacts
+end_time: 1621790754185
+entry_point_name: ''
+experiment_id: '0'
+lifecycle_stage: active
+name: ''
+run_id: 794e3e4e628e416289e6237f26ef142d
+run_uuid: 794e3e4e628e416289e6237f26ef142d
+source_name: ''
+source_type: 4
+source_version: ''
+start_time: 1621790714349
+status: 3
+tags: []
+user_id: tomasz
diff --git a/mlruns/0/794e3e4e628e416289e6237f26ef142d/metrics/rmse b/mlruns/0/794e3e4e628e416289e6237f26ef142d/metrics/rmse
new file mode 100644
index 0000000..478ab20
--- /dev/null
+++ b/mlruns/0/794e3e4e628e416289e6237f26ef142d/metrics/rmse
@@ -0,0 +1 @@
+1621790752218 0.031056313979876536 0
diff --git a/mlruns/0/794e3e4e628e416289e6237f26ef142d/params/batch_size b/mlruns/0/794e3e4e628e416289e6237f26ef142d/params/batch_size
new file mode 100644
index 0000000..19c7bdb
--- /dev/null
+++ b/mlruns/0/794e3e4e628e416289e6237f26ef142d/params/batch_size
@@ -0,0 +1 @@
+16
\ No newline at end of file
diff --git a/mlruns/0/794e3e4e628e416289e6237f26ef142d/params/epochs b/mlruns/0/794e3e4e628e416289e6237f26ef142d/params/epochs
new file mode 100644
index 0000000..3f10ffe
--- /dev/null
+++ b/mlruns/0/794e3e4e628e416289e6237f26ef142d/params/epochs
@@ -0,0 +1 @@
+15
\ No newline at end of file
diff --git a/mlruns/0/794e3e4e628e416289e6237f26ef142d/tags/mlflow.source.git.commit b/mlruns/0/794e3e4e628e416289e6237f26ef142d/tags/mlflow.source.git.commit
new file mode 100644
index 0000000..2e3fdf7
--- /dev/null
+++ b/mlruns/0/794e3e4e628e416289e6237f26ef142d/tags/mlflow.source.git.commit
@@ -0,0 +1 @@
+db58b281061f170a706135380f1e7c86c7162acc
\ No newline at end of file
diff --git a/mlruns/0/794e3e4e628e416289e6237f26ef142d/tags/mlflow.source.name b/mlruns/0/794e3e4e628e416289e6237f26ef142d/tags/mlflow.source.name
new file mode 100644
index 0000000..939bfd8
--- /dev/null
+++ b/mlruns/0/794e3e4e628e416289e6237f26ef142d/tags/mlflow.source.name
@@ -0,0 +1 @@
+/home/tomasz/IUM/ium_434695/vgsales-mlflow.py
\ No newline at end of file
diff --git a/mlruns/0/794e3e4e628e416289e6237f26ef142d/tags/mlflow.source.type b/mlruns/0/794e3e4e628e416289e6237f26ef142d/tags/mlflow.source.type
new file mode 100644
index 0000000..0c2c1fe
--- /dev/null
+++ b/mlruns/0/794e3e4e628e416289e6237f26ef142d/tags/mlflow.source.type
@@ -0,0 +1 @@
+LOCAL
\ No newline at end of file
diff --git a/mlruns/0/794e3e4e628e416289e6237f26ef142d/tags/mlflow.user b/mlruns/0/794e3e4e628e416289e6237f26ef142d/tags/mlflow.user
new file mode 100644
index 0000000..4b82f83
--- /dev/null
+++ b/mlruns/0/794e3e4e628e416289e6237f26ef142d/tags/mlflow.user
@@ -0,0 +1 @@
+tomasz
\ No newline at end of file
diff --git a/mlruns/0/883a0165bd624848bff8037bbad39446/artifacts/vgsales_model.h5/MLmodel b/mlruns/0/883a0165bd624848bff8037bbad39446/artifacts/vgsales_model.h5/MLmodel
new file mode 100644
index 0000000..12413c5
--- /dev/null
+++ b/mlruns/0/883a0165bd624848bff8037bbad39446/artifacts/vgsales_model.h5/MLmodel
@@ -0,0 +1,14 @@
+artifact_path: vgsales_model.h5
+flavors:
+  keras:
+    data: data
+    keras_module: tensorflow.keras
+    keras_version: 2.4.0
+    save_format: tf
+  python_function:
+    data: data
+    env: conda.yaml
+    loader_module: mlflow.keras
+    python_version: 3.8.5
+run_id: 883a0165bd624848bff8037bbad39446
+utc_time_created: '2021-05-23 17:12:20.066683'
diff --git a/mlruns/0/883a0165bd624848bff8037bbad39446/artifacts/vgsales_model.h5/conda.yaml b/mlruns/0/883a0165bd624848bff8037bbad39446/artifacts/vgsales_model.h5/conda.yaml
new file mode 100644
index 0000000..3fa4018
--- /dev/null
+++ b/mlruns/0/883a0165bd624848bff8037bbad39446/artifacts/vgsales_model.h5/conda.yaml
@@ -0,0 +1,10 @@
+channels:
+- defaults
+- conda-forge
+dependencies:
+- python=3.8.5
+- pip
+- pip:
+  - mlflow
+  - tensorflow==2.4.1
+name: mlflow-env
diff --git a/mlruns/0/883a0165bd624848bff8037bbad39446/artifacts/vgsales_model.h5/data/keras_module.txt b/mlruns/0/883a0165bd624848bff8037bbad39446/artifacts/vgsales_model.h5/data/keras_module.txt
new file mode 100644
index 0000000..2c73dfd
--- /dev/null
+++ b/mlruns/0/883a0165bd624848bff8037bbad39446/artifacts/vgsales_model.h5/data/keras_module.txt
@@ -0,0 +1 @@
+tensorflow.keras
\ No newline at end of file
diff --git a/mlruns/0/883a0165bd624848bff8037bbad39446/artifacts/vgsales_model.h5/data/model/saved_model.pb b/mlruns/0/883a0165bd624848bff8037bbad39446/artifacts/vgsales_model.h5/data/model/saved_model.pb
new file mode 100644
index 0000000..dabd46e
Binary files /dev/null and b/mlruns/0/883a0165bd624848bff8037bbad39446/artifacts/vgsales_model.h5/data/model/saved_model.pb differ
diff --git a/mlruns/0/883a0165bd624848bff8037bbad39446/artifacts/vgsales_model.h5/data/model/variables/variables.data-00000-of-00001 b/mlruns/0/883a0165bd624848bff8037bbad39446/artifacts/vgsales_model.h5/data/model/variables/variables.data-00000-of-00001
new file mode 100644
index 0000000..b208f8d
Binary files /dev/null and b/mlruns/0/883a0165bd624848bff8037bbad39446/artifacts/vgsales_model.h5/data/model/variables/variables.data-00000-of-00001 differ
diff --git a/mlruns/0/883a0165bd624848bff8037bbad39446/artifacts/vgsales_model.h5/data/model/variables/variables.index b/mlruns/0/883a0165bd624848bff8037bbad39446/artifacts/vgsales_model.h5/data/model/variables/variables.index
new file mode 100644
index 0000000..cac950b
Binary files /dev/null and b/mlruns/0/883a0165bd624848bff8037bbad39446/artifacts/vgsales_model.h5/data/model/variables/variables.index differ
diff --git a/mlruns/0/883a0165bd624848bff8037bbad39446/artifacts/vgsales_model.h5/data/save_format.txt b/mlruns/0/883a0165bd624848bff8037bbad39446/artifacts/vgsales_model.h5/data/save_format.txt
new file mode 100644
index 0000000..439ab29
--- /dev/null
+++ b/mlruns/0/883a0165bd624848bff8037bbad39446/artifacts/vgsales_model.h5/data/save_format.txt
@@ -0,0 +1 @@
+tf
\ No newline at end of file
diff --git a/mlruns/0/883a0165bd624848bff8037bbad39446/meta.yaml b/mlruns/0/883a0165bd624848bff8037bbad39446/meta.yaml
new file mode 100644
index 0000000..0858eed
--- /dev/null
+++ b/mlruns/0/883a0165bd624848bff8037bbad39446/meta.yaml
@@ -0,0 +1,15 @@
+artifact_uri: file:///home/tomasz/IUM/ium_434695/mlruns/0/883a0165bd624848bff8037bbad39446/artifacts
+end_time: 1621789941879
+entry_point_name: ''
+experiment_id: '0'
+lifecycle_stage: active
+name: ''
+run_id: 883a0165bd624848bff8037bbad39446
+run_uuid: 883a0165bd624848bff8037bbad39446
+source_name: ''
+source_type: 4
+source_version: ''
+start_time: 1621789909395
+status: 3
+tags: []
+user_id: tomasz
diff --git a/mlruns/0/883a0165bd624848bff8037bbad39446/metrics/rmse b/mlruns/0/883a0165bd624848bff8037bbad39446/metrics/rmse
new file mode 100644
index 0000000..535d2bf
--- /dev/null
+++ b/mlruns/0/883a0165bd624848bff8037bbad39446/metrics/rmse
@@ -0,0 +1 @@
+1621789940064 0.03283662485627263 0
diff --git a/mlruns/0/883a0165bd624848bff8037bbad39446/params/batch_size b/mlruns/0/883a0165bd624848bff8037bbad39446/params/batch_size
new file mode 100644
index 0000000..19c7bdb
--- /dev/null
+++ b/mlruns/0/883a0165bd624848bff8037bbad39446/params/batch_size
@@ -0,0 +1 @@
+16
\ No newline at end of file
diff --git a/mlruns/0/883a0165bd624848bff8037bbad39446/params/epochs b/mlruns/0/883a0165bd624848bff8037bbad39446/params/epochs
new file mode 100644
index 0000000..3f10ffe
--- /dev/null
+++ b/mlruns/0/883a0165bd624848bff8037bbad39446/params/epochs
@@ -0,0 +1 @@
+15
\ No newline at end of file
diff --git a/mlruns/0/883a0165bd624848bff8037bbad39446/tags/mlflow.log-model.history b/mlruns/0/883a0165bd624848bff8037bbad39446/tags/mlflow.log-model.history
new file mode 100644
index 0000000..d0de125
--- /dev/null
+++ b/mlruns/0/883a0165bd624848bff8037bbad39446/tags/mlflow.log-model.history
@@ -0,0 +1 @@
+[{"run_id": "883a0165bd624848bff8037bbad39446", "artifact_path": "vgsales_model.h5", "utc_time_created": "2021-05-23 17:12:20.066683", "flavors": {"keras": {"keras_module": "tensorflow.keras", "keras_version": "2.4.0", "save_format": "tf", "data": "data"}, "python_function": {"loader_module": "mlflow.keras", "python_version": "3.8.5", "data": "data", "env": "conda.yaml"}}}]
\ No newline at end of file
diff --git a/mlruns/0/883a0165bd624848bff8037bbad39446/tags/mlflow.source.git.commit b/mlruns/0/883a0165bd624848bff8037bbad39446/tags/mlflow.source.git.commit
new file mode 100644
index 0000000..2e3fdf7
--- /dev/null
+++ b/mlruns/0/883a0165bd624848bff8037bbad39446/tags/mlflow.source.git.commit
@@ -0,0 +1 @@
+db58b281061f170a706135380f1e7c86c7162acc
\ No newline at end of file
diff --git a/mlruns/0/883a0165bd624848bff8037bbad39446/tags/mlflow.source.name b/mlruns/0/883a0165bd624848bff8037bbad39446/tags/mlflow.source.name
new file mode 100644
index 0000000..939bfd8
--- /dev/null
+++ b/mlruns/0/883a0165bd624848bff8037bbad39446/tags/mlflow.source.name
@@ -0,0 +1 @@
+/home/tomasz/IUM/ium_434695/vgsales-mlflow.py
\ No newline at end of file
diff --git a/mlruns/0/883a0165bd624848bff8037bbad39446/tags/mlflow.source.type b/mlruns/0/883a0165bd624848bff8037bbad39446/tags/mlflow.source.type
new file mode 100644
index 0000000..0c2c1fe
--- /dev/null
+++ b/mlruns/0/883a0165bd624848bff8037bbad39446/tags/mlflow.source.type
@@ -0,0 +1 @@
+LOCAL
\ No newline at end of file
diff --git a/mlruns/0/883a0165bd624848bff8037bbad39446/tags/mlflow.user b/mlruns/0/883a0165bd624848bff8037bbad39446/tags/mlflow.user
new file mode 100644
index 0000000..4b82f83
--- /dev/null
+++ b/mlruns/0/883a0165bd624848bff8037bbad39446/tags/mlflow.user
@@ -0,0 +1 @@
+tomasz
\ No newline at end of file
diff --git a/mlruns/0/c982ec1561894780b45c04392e54dede/artifacts/vgsales_model.h5/MLmodel b/mlruns/0/c982ec1561894780b45c04392e54dede/artifacts/vgsales_model.h5/MLmodel
new file mode 100644
index 0000000..4593833
--- /dev/null
+++ b/mlruns/0/c982ec1561894780b45c04392e54dede/artifacts/vgsales_model.h5/MLmodel
@@ -0,0 +1,14 @@
+artifact_path: vgsales_model.h5
+flavors:
+  keras:
+    data: data
+    keras_module: tensorflow.keras
+    keras_version: 2.4.0
+    save_format: tf
+  python_function:
+    data: data
+    env: conda.yaml
+    loader_module: mlflow.keras
+    python_version: 3.8.5
+run_id: c982ec1561894780b45c04392e54dede
+utc_time_created: '2021-05-23 17:19:52.104292'
diff --git a/mlruns/0/c982ec1561894780b45c04392e54dede/artifacts/vgsales_model.h5/conda.yaml b/mlruns/0/c982ec1561894780b45c04392e54dede/artifacts/vgsales_model.h5/conda.yaml
new file mode 100644
index 0000000..3fa4018
--- /dev/null
+++ b/mlruns/0/c982ec1561894780b45c04392e54dede/artifacts/vgsales_model.h5/conda.yaml
@@ -0,0 +1,10 @@
+channels:
+- defaults
+- conda-forge
+dependencies:
+- python=3.8.5
+- pip
+- pip:
+  - mlflow
+  - tensorflow==2.4.1
+name: mlflow-env
diff --git a/mlruns/0/c982ec1561894780b45c04392e54dede/artifacts/vgsales_model.h5/data/keras_module.txt b/mlruns/0/c982ec1561894780b45c04392e54dede/artifacts/vgsales_model.h5/data/keras_module.txt
new file mode 100644
index 0000000..2c73dfd
--- /dev/null
+++ b/mlruns/0/c982ec1561894780b45c04392e54dede/artifacts/vgsales_model.h5/data/keras_module.txt
@@ -0,0 +1 @@
+tensorflow.keras
\ No newline at end of file
diff --git a/mlruns/0/c982ec1561894780b45c04392e54dede/artifacts/vgsales_model.h5/data/model/saved_model.pb b/mlruns/0/c982ec1561894780b45c04392e54dede/artifacts/vgsales_model.h5/data/model/saved_model.pb
new file mode 100644
index 0000000..f8132f7
Binary files /dev/null and b/mlruns/0/c982ec1561894780b45c04392e54dede/artifacts/vgsales_model.h5/data/model/saved_model.pb differ
diff --git a/mlruns/0/c982ec1561894780b45c04392e54dede/artifacts/vgsales_model.h5/data/model/variables/variables.data-00000-of-00001 b/mlruns/0/c982ec1561894780b45c04392e54dede/artifacts/vgsales_model.h5/data/model/variables/variables.data-00000-of-00001
new file mode 100644
index 0000000..78901c7
Binary files /dev/null and b/mlruns/0/c982ec1561894780b45c04392e54dede/artifacts/vgsales_model.h5/data/model/variables/variables.data-00000-of-00001 differ
diff --git a/mlruns/0/c982ec1561894780b45c04392e54dede/artifacts/vgsales_model.h5/data/model/variables/variables.index b/mlruns/0/c982ec1561894780b45c04392e54dede/artifacts/vgsales_model.h5/data/model/variables/variables.index
new file mode 100644
index 0000000..ba43e30
Binary files /dev/null and b/mlruns/0/c982ec1561894780b45c04392e54dede/artifacts/vgsales_model.h5/data/model/variables/variables.index differ
diff --git a/mlruns/0/c982ec1561894780b45c04392e54dede/artifacts/vgsales_model.h5/data/save_format.txt b/mlruns/0/c982ec1561894780b45c04392e54dede/artifacts/vgsales_model.h5/data/save_format.txt
new file mode 100644
index 0000000..439ab29
--- /dev/null
+++ b/mlruns/0/c982ec1561894780b45c04392e54dede/artifacts/vgsales_model.h5/data/save_format.txt
@@ -0,0 +1 @@
+tf
\ No newline at end of file
diff --git a/mlruns/0/c982ec1561894780b45c04392e54dede/meta.yaml b/mlruns/0/c982ec1561894780b45c04392e54dede/meta.yaml
new file mode 100644
index 0000000..70ef3db
--- /dev/null
+++ b/mlruns/0/c982ec1561894780b45c04392e54dede/meta.yaml
@@ -0,0 +1,15 @@
+artifact_uri: file:///home/tomasz/IUM/ium_434695/mlruns/0/c982ec1561894780b45c04392e54dede/artifacts
+end_time: 1621790393452
+entry_point_name: ''
+experiment_id: '0'
+lifecycle_stage: active
+name: ''
+run_id: c982ec1561894780b45c04392e54dede
+run_uuid: c982ec1561894780b45c04392e54dede
+source_name: ''
+source_type: 4
+source_version: ''
+start_time: 1621790353667
+status: 4
+tags: []
+user_id: tomasz
diff --git a/mlruns/0/c982ec1561894780b45c04392e54dede/metrics/rmse b/mlruns/0/c982ec1561894780b45c04392e54dede/metrics/rmse
new file mode 100644
index 0000000..f6db58c
--- /dev/null
+++ b/mlruns/0/c982ec1561894780b45c04392e54dede/metrics/rmse
@@ -0,0 +1 @@
+1621790392100 0.03209544018660911 0
diff --git a/mlruns/0/c982ec1561894780b45c04392e54dede/params/batch_size b/mlruns/0/c982ec1561894780b45c04392e54dede/params/batch_size
new file mode 100644
index 0000000..19c7bdb
--- /dev/null
+++ b/mlruns/0/c982ec1561894780b45c04392e54dede/params/batch_size
@@ -0,0 +1 @@
+16
\ No newline at end of file
diff --git a/mlruns/0/c982ec1561894780b45c04392e54dede/params/epochs b/mlruns/0/c982ec1561894780b45c04392e54dede/params/epochs
new file mode 100644
index 0000000..3f10ffe
--- /dev/null
+++ b/mlruns/0/c982ec1561894780b45c04392e54dede/params/epochs
@@ -0,0 +1 @@
+15
\ No newline at end of file
diff --git a/mlruns/0/c982ec1561894780b45c04392e54dede/tags/mlflow.log-model.history b/mlruns/0/c982ec1561894780b45c04392e54dede/tags/mlflow.log-model.history
new file mode 100644
index 0000000..9b81d37
--- /dev/null
+++ b/mlruns/0/c982ec1561894780b45c04392e54dede/tags/mlflow.log-model.history
@@ -0,0 +1 @@
+[{"run_id": "c982ec1561894780b45c04392e54dede", "artifact_path": "vgsales_model.h5", "utc_time_created": "2021-05-23 17:19:52.104292", "flavors": {"keras": {"keras_module": "tensorflow.keras", "keras_version": "2.4.0", "save_format": "tf", "data": "data"}, "python_function": {"loader_module": "mlflow.keras", "python_version": "3.8.5", "data": "data", "env": "conda.yaml"}}}]
\ No newline at end of file
diff --git a/mlruns/0/c982ec1561894780b45c04392e54dede/tags/mlflow.source.git.commit b/mlruns/0/c982ec1561894780b45c04392e54dede/tags/mlflow.source.git.commit
new file mode 100644
index 0000000..2e3fdf7
--- /dev/null
+++ b/mlruns/0/c982ec1561894780b45c04392e54dede/tags/mlflow.source.git.commit
@@ -0,0 +1 @@
+db58b281061f170a706135380f1e7c86c7162acc
\ No newline at end of file
diff --git a/mlruns/0/c982ec1561894780b45c04392e54dede/tags/mlflow.source.name b/mlruns/0/c982ec1561894780b45c04392e54dede/tags/mlflow.source.name
new file mode 100644
index 0000000..939bfd8
--- /dev/null
+++ b/mlruns/0/c982ec1561894780b45c04392e54dede/tags/mlflow.source.name
@@ -0,0 +1 @@
+/home/tomasz/IUM/ium_434695/vgsales-mlflow.py
\ No newline at end of file
diff --git a/mlruns/0/c982ec1561894780b45c04392e54dede/tags/mlflow.source.type b/mlruns/0/c982ec1561894780b45c04392e54dede/tags/mlflow.source.type
new file mode 100644
index 0000000..0c2c1fe
--- /dev/null
+++ b/mlruns/0/c982ec1561894780b45c04392e54dede/tags/mlflow.source.type
@@ -0,0 +1 @@
+LOCAL
\ No newline at end of file
diff --git a/mlruns/0/c982ec1561894780b45c04392e54dede/tags/mlflow.user b/mlruns/0/c982ec1561894780b45c04392e54dede/tags/mlflow.user
new file mode 100644
index 0000000..4b82f83
--- /dev/null
+++ b/mlruns/0/c982ec1561894780b45c04392e54dede/tags/mlflow.user
@@ -0,0 +1 @@
+tomasz
\ No newline at end of file
diff --git a/mlruns/0/d981b8d00a9240d585647c760d4143a3/meta.yaml b/mlruns/0/d981b8d00a9240d585647c760d4143a3/meta.yaml
new file mode 100644
index 0000000..1c8f61d
--- /dev/null
+++ b/mlruns/0/d981b8d00a9240d585647c760d4143a3/meta.yaml
@@ -0,0 +1,15 @@
+artifact_uri: file:///home/tomasz/IUM/ium_434695/mlruns/0/d981b8d00a9240d585647c760d4143a3/artifacts
+end_time: 1621790667043
+entry_point_name: ''
+experiment_id: '0'
+lifecycle_stage: active
+name: ''
+run_id: d981b8d00a9240d585647c760d4143a3
+run_uuid: d981b8d00a9240d585647c760d4143a3
+source_name: ''
+source_type: 4
+source_version: ''
+start_time: 1621790630159
+status: 4
+tags: []
+user_id: tomasz
diff --git a/mlruns/0/d981b8d00a9240d585647c760d4143a3/metrics/rmse b/mlruns/0/d981b8d00a9240d585647c760d4143a3/metrics/rmse
new file mode 100644
index 0000000..28175a8
--- /dev/null
+++ b/mlruns/0/d981b8d00a9240d585647c760d4143a3/metrics/rmse
@@ -0,0 +1 @@
+1621790667036 0.032142209216265996 0
diff --git a/mlruns/0/d981b8d00a9240d585647c760d4143a3/params/batch_size b/mlruns/0/d981b8d00a9240d585647c760d4143a3/params/batch_size
new file mode 100644
index 0000000..19c7bdb
--- /dev/null
+++ b/mlruns/0/d981b8d00a9240d585647c760d4143a3/params/batch_size
@@ -0,0 +1 @@
+16
\ No newline at end of file
diff --git a/mlruns/0/d981b8d00a9240d585647c760d4143a3/params/epochs b/mlruns/0/d981b8d00a9240d585647c760d4143a3/params/epochs
new file mode 100644
index 0000000..3f10ffe
--- /dev/null
+++ b/mlruns/0/d981b8d00a9240d585647c760d4143a3/params/epochs
@@ -0,0 +1 @@
+15
\ No newline at end of file
diff --git a/mlruns/0/d981b8d00a9240d585647c760d4143a3/tags/mlflow.source.git.commit b/mlruns/0/d981b8d00a9240d585647c760d4143a3/tags/mlflow.source.git.commit
new file mode 100644
index 0000000..2e3fdf7
--- /dev/null
+++ b/mlruns/0/d981b8d00a9240d585647c760d4143a3/tags/mlflow.source.git.commit
@@ -0,0 +1 @@
+db58b281061f170a706135380f1e7c86c7162acc
\ No newline at end of file
diff --git a/mlruns/0/d981b8d00a9240d585647c760d4143a3/tags/mlflow.source.name b/mlruns/0/d981b8d00a9240d585647c760d4143a3/tags/mlflow.source.name
new file mode 100644
index 0000000..939bfd8
--- /dev/null
+++ b/mlruns/0/d981b8d00a9240d585647c760d4143a3/tags/mlflow.source.name
@@ -0,0 +1 @@
+/home/tomasz/IUM/ium_434695/vgsales-mlflow.py
\ No newline at end of file
diff --git a/mlruns/0/d981b8d00a9240d585647c760d4143a3/tags/mlflow.source.type b/mlruns/0/d981b8d00a9240d585647c760d4143a3/tags/mlflow.source.type
new file mode 100644
index 0000000..0c2c1fe
--- /dev/null
+++ b/mlruns/0/d981b8d00a9240d585647c760d4143a3/tags/mlflow.source.type
@@ -0,0 +1 @@
+LOCAL
\ No newline at end of file
diff --git a/mlruns/0/d981b8d00a9240d585647c760d4143a3/tags/mlflow.user b/mlruns/0/d981b8d00a9240d585647c760d4143a3/tags/mlflow.user
new file mode 100644
index 0000000..4b82f83
--- /dev/null
+++ b/mlruns/0/d981b8d00a9240d585647c760d4143a3/tags/mlflow.user
@@ -0,0 +1 @@
+tomasz
\ No newline at end of file
diff --git a/mlruns/0/fc71b79472444c8eb0ff1854dd477712/meta.yaml b/mlruns/0/fc71b79472444c8eb0ff1854dd477712/meta.yaml
new file mode 100644
index 0000000..f7522c4
--- /dev/null
+++ b/mlruns/0/fc71b79472444c8eb0ff1854dd477712/meta.yaml
@@ -0,0 +1,15 @@
+artifact_uri: file:///home/tomasz/IUM/ium_434695/mlruns/0/fc71b79472444c8eb0ff1854dd477712/artifacts
+end_time: 1621790569106
+entry_point_name: ''
+experiment_id: '0'
+lifecycle_stage: active
+name: ''
+run_id: fc71b79472444c8eb0ff1854dd477712
+run_uuid: fc71b79472444c8eb0ff1854dd477712
+source_name: ''
+source_type: 4
+source_version: ''
+start_time: 1621790533743
+status: 4
+tags: []
+user_id: tomasz
diff --git a/mlruns/0/fc71b79472444c8eb0ff1854dd477712/metrics/rmse b/mlruns/0/fc71b79472444c8eb0ff1854dd477712/metrics/rmse
new file mode 100644
index 0000000..4e599f3
--- /dev/null
+++ b/mlruns/0/fc71b79472444c8eb0ff1854dd477712/metrics/rmse
@@ -0,0 +1 @@
+1621790569099 0.032763813909841656 0
diff --git a/mlruns/0/fc71b79472444c8eb0ff1854dd477712/params/batch_size b/mlruns/0/fc71b79472444c8eb0ff1854dd477712/params/batch_size
new file mode 100644
index 0000000..19c7bdb
--- /dev/null
+++ b/mlruns/0/fc71b79472444c8eb0ff1854dd477712/params/batch_size
@@ -0,0 +1 @@
+16
\ No newline at end of file
diff --git a/mlruns/0/fc71b79472444c8eb0ff1854dd477712/params/epochs b/mlruns/0/fc71b79472444c8eb0ff1854dd477712/params/epochs
new file mode 100644
index 0000000..3f10ffe
--- /dev/null
+++ b/mlruns/0/fc71b79472444c8eb0ff1854dd477712/params/epochs
@@ -0,0 +1 @@
+15
\ No newline at end of file
diff --git a/mlruns/0/fc71b79472444c8eb0ff1854dd477712/tags/mlflow.source.git.commit b/mlruns/0/fc71b79472444c8eb0ff1854dd477712/tags/mlflow.source.git.commit
new file mode 100644
index 0000000..2e3fdf7
--- /dev/null
+++ b/mlruns/0/fc71b79472444c8eb0ff1854dd477712/tags/mlflow.source.git.commit
@@ -0,0 +1 @@
+db58b281061f170a706135380f1e7c86c7162acc
\ No newline at end of file
diff --git a/mlruns/0/fc71b79472444c8eb0ff1854dd477712/tags/mlflow.source.name b/mlruns/0/fc71b79472444c8eb0ff1854dd477712/tags/mlflow.source.name
new file mode 100644
index 0000000..939bfd8
--- /dev/null
+++ b/mlruns/0/fc71b79472444c8eb0ff1854dd477712/tags/mlflow.source.name
@@ -0,0 +1 @@
+/home/tomasz/IUM/ium_434695/vgsales-mlflow.py
\ No newline at end of file
diff --git a/mlruns/0/fc71b79472444c8eb0ff1854dd477712/tags/mlflow.source.type b/mlruns/0/fc71b79472444c8eb0ff1854dd477712/tags/mlflow.source.type
new file mode 100644
index 0000000..0c2c1fe
--- /dev/null
+++ b/mlruns/0/fc71b79472444c8eb0ff1854dd477712/tags/mlflow.source.type
@@ -0,0 +1 @@
+LOCAL
\ No newline at end of file
diff --git a/mlruns/0/fc71b79472444c8eb0ff1854dd477712/tags/mlflow.user b/mlruns/0/fc71b79472444c8eb0ff1854dd477712/tags/mlflow.user
new file mode 100644
index 0000000..4b82f83
--- /dev/null
+++ b/mlruns/0/fc71b79472444c8eb0ff1854dd477712/tags/mlflow.user
@@ -0,0 +1 @@
+tomasz
\ No newline at end of file
diff --git a/mlruns/0/meta.yaml b/mlruns/0/meta.yaml
new file mode 100644
index 0000000..8a02dc3
--- /dev/null
+++ b/mlruns/0/meta.yaml
@@ -0,0 +1,4 @@
+artifact_location: file:///home/tomasz/IUM/ium_434695/mlruns/0
+experiment_id: '0'
+lifecycle_stage: active
+name: Default
diff --git a/vgsales-mlflow.py b/vgsales-mlflow.py
index bbc599b..57add00 100644
--- a/vgsales-mlflow.py
+++ b/vgsales-mlflow.py
@@ -49,7 +49,7 @@ def my_main(epochs, batch_size):
 
     model.save('vgsales_model.h5')
 
-    return rmse, model
+    return rmse, model, X_train, y_train
 
 
 
@@ -59,10 +59,11 @@ batch_size = int(sys.argv[2]) if len(sys.argv) > 2 else 16
 
 with mlflow.start_run():
         
-    rmse, model = my_main(epochs, batch_size)
+    rmse, model, x_train, y_train = my_main(epochs, batch_size)
 
     mlflow.log_param("epochs", epochs)
     mlflow.log_param("batch_size", batch_size)
     mlflow.log_metric("rmse", rmse) 
-    mlflow.keras.log_model(model, 'vgsales_model.h5')
-    mlflow.keras.save_model(model, "my_model")
\ No newline at end of file
+    #mlflow.keras.log_model(model, 'vgsales_model.h5')
+    mlflow.keras.save_model(model, "my_model", signature=mlflow.models.signature.infer_signature(x_train, y_train), input_example=x_train)
+    
\ No newline at end of file
diff --git a/vgsales_model.h5 b/vgsales_model.h5
new file mode 100644
index 0000000..be4f360
Binary files /dev/null and b/vgsales_model.h5 differ