diff --git a/mlruns/0/156c624fa6ff468887bf9543a37ce8b3/meta.yaml b/mlruns/0/156c624fa6ff468887bf9543a37ce8b3/meta.yaml new file mode 100644 index 0000000..f9591e3 --- /dev/null +++ b/mlruns/0/156c624fa6ff468887bf9543a37ce8b3/meta.yaml @@ -0,0 +1,15 @@ +artifact_uri: file:///home/tomasz/IUM/ium_434695/mlruns/0/156c624fa6ff468887bf9543a37ce8b3/artifacts +end_time: 1621804324404 +entry_point_name: '' +experiment_id: '0' +lifecycle_stage: active +name: '' +run_id: 156c624fa6ff468887bf9543a37ce8b3 +run_uuid: 156c624fa6ff468887bf9543a37ce8b3 +source_name: '' +source_type: 4 +source_version: '' +start_time: 1621804324311 +status: 4 +tags: [] +user_id: tomasz diff --git a/mlruns/0/156c624fa6ff468887bf9543a37ce8b3/tags/mlflow.source.git.commit b/mlruns/0/156c624fa6ff468887bf9543a37ce8b3/tags/mlflow.source.git.commit new file mode 100644 index 0000000..5ecba19 --- /dev/null +++ b/mlruns/0/156c624fa6ff468887bf9543a37ce8b3/tags/mlflow.source.git.commit @@ -0,0 +1 @@ +b5774c67b74e143c2e0d3a990180a7305140b234 \ No newline at end of file diff --git a/mlruns/0/156c624fa6ff468887bf9543a37ce8b3/tags/mlflow.source.name b/mlruns/0/156c624fa6ff468887bf9543a37ce8b3/tags/mlflow.source.name new file mode 100644 index 0000000..939bfd8 --- /dev/null +++ b/mlruns/0/156c624fa6ff468887bf9543a37ce8b3/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/156c624fa6ff468887bf9543a37ce8b3/tags/mlflow.source.type b/mlruns/0/156c624fa6ff468887bf9543a37ce8b3/tags/mlflow.source.type new file mode 100644 index 0000000..0c2c1fe --- /dev/null +++ b/mlruns/0/156c624fa6ff468887bf9543a37ce8b3/tags/mlflow.source.type @@ -0,0 +1 @@ +LOCAL \ No newline at end of file diff --git a/mlruns/0/156c624fa6ff468887bf9543a37ce8b3/tags/mlflow.user b/mlruns/0/156c624fa6ff468887bf9543a37ce8b3/tags/mlflow.user new file mode 100644 index 0000000..4b82f83 --- /dev/null +++ b/mlruns/0/156c624fa6ff468887bf9543a37ce8b3/tags/mlflow.user @@ -0,0 +1 @@ +tomasz \ No newline at end of file diff --git a/mlruns/0/4be4d7d988dc477e8ec483ca0ec5bb30/meta.yaml b/mlruns/0/4be4d7d988dc477e8ec483ca0ec5bb30/meta.yaml new file mode 100644 index 0000000..2ab2d15 --- /dev/null +++ b/mlruns/0/4be4d7d988dc477e8ec483ca0ec5bb30/meta.yaml @@ -0,0 +1,15 @@ +artifact_uri: file:///home/tomasz/IUM/ium_434695/mlruns/0/4be4d7d988dc477e8ec483ca0ec5bb30/artifacts +end_time: null +entry_point_name: '' +experiment_id: '0' +lifecycle_stage: active +name: '' +run_id: 4be4d7d988dc477e8ec483ca0ec5bb30 +run_uuid: 4be4d7d988dc477e8ec483ca0ec5bb30 +source_name: '' +source_type: 4 +source_version: '' +start_time: 1621804433172 +status: 1 +tags: [] +user_id: tomasz diff --git a/mlruns/0/4be4d7d988dc477e8ec483ca0ec5bb30/tags/mlflow.source.git.commit b/mlruns/0/4be4d7d988dc477e8ec483ca0ec5bb30/tags/mlflow.source.git.commit new file mode 100644 index 0000000..5ecba19 --- /dev/null +++ b/mlruns/0/4be4d7d988dc477e8ec483ca0ec5bb30/tags/mlflow.source.git.commit @@ -0,0 +1 @@ +b5774c67b74e143c2e0d3a990180a7305140b234 \ No newline at end of file diff --git a/mlruns/0/4be4d7d988dc477e8ec483ca0ec5bb30/tags/mlflow.source.name b/mlruns/0/4be4d7d988dc477e8ec483ca0ec5bb30/tags/mlflow.source.name new file mode 100644 index 0000000..939bfd8 --- /dev/null +++ b/mlruns/0/4be4d7d988dc477e8ec483ca0ec5bb30/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/4be4d7d988dc477e8ec483ca0ec5bb30/tags/mlflow.source.type b/mlruns/0/4be4d7d988dc477e8ec483ca0ec5bb30/tags/mlflow.source.type new file mode 100644 index 0000000..0c2c1fe --- /dev/null +++ b/mlruns/0/4be4d7d988dc477e8ec483ca0ec5bb30/tags/mlflow.source.type @@ -0,0 +1 @@ +LOCAL \ No newline at end of file diff --git a/mlruns/0/4be4d7d988dc477e8ec483ca0ec5bb30/tags/mlflow.user b/mlruns/0/4be4d7d988dc477e8ec483ca0ec5bb30/tags/mlflow.user new file mode 100644 index 0000000..4b82f83 --- /dev/null +++ b/mlruns/0/4be4d7d988dc477e8ec483ca0ec5bb30/tags/mlflow.user @@ -0,0 +1 @@ +tomasz \ No newline at end of file diff --git a/mlruns/0/b19dc8b40ada4fdea65402cf2f8a4c39/meta.yaml b/mlruns/0/b19dc8b40ada4fdea65402cf2f8a4c39/meta.yaml new file mode 100644 index 0000000..b071c72 --- /dev/null +++ b/mlruns/0/b19dc8b40ada4fdea65402cf2f8a4c39/meta.yaml @@ -0,0 +1,15 @@ +artifact_uri: file:///home/tomasz/IUM/ium_434695/mlruns/0/b19dc8b40ada4fdea65402cf2f8a4c39/artifacts +end_time: 1621804368194 +entry_point_name: '' +experiment_id: '0' +lifecycle_stage: active +name: '' +run_id: b19dc8b40ada4fdea65402cf2f8a4c39 +run_uuid: b19dc8b40ada4fdea65402cf2f8a4c39 +source_name: '' +source_type: 4 +source_version: '' +start_time: 1621804367588 +status: 4 +tags: [] +user_id: tomasz diff --git a/mlruns/0/b19dc8b40ada4fdea65402cf2f8a4c39/tags/mlflow.source.git.commit b/mlruns/0/b19dc8b40ada4fdea65402cf2f8a4c39/tags/mlflow.source.git.commit new file mode 100644 index 0000000..5ecba19 --- /dev/null +++ b/mlruns/0/b19dc8b40ada4fdea65402cf2f8a4c39/tags/mlflow.source.git.commit @@ -0,0 +1 @@ +b5774c67b74e143c2e0d3a990180a7305140b234 \ No newline at end of file diff --git a/mlruns/0/b19dc8b40ada4fdea65402cf2f8a4c39/tags/mlflow.source.name b/mlruns/0/b19dc8b40ada4fdea65402cf2f8a4c39/tags/mlflow.source.name new file mode 100644 index 0000000..939bfd8 --- /dev/null +++ b/mlruns/0/b19dc8b40ada4fdea65402cf2f8a4c39/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/b19dc8b40ada4fdea65402cf2f8a4c39/tags/mlflow.source.type b/mlruns/0/b19dc8b40ada4fdea65402cf2f8a4c39/tags/mlflow.source.type new file mode 100644 index 0000000..0c2c1fe --- /dev/null +++ b/mlruns/0/b19dc8b40ada4fdea65402cf2f8a4c39/tags/mlflow.source.type @@ -0,0 +1 @@ +LOCAL \ No newline at end of file diff --git a/mlruns/0/b19dc8b40ada4fdea65402cf2f8a4c39/tags/mlflow.user b/mlruns/0/b19dc8b40ada4fdea65402cf2f8a4c39/tags/mlflow.user new file mode 100644 index 0000000..4b82f83 --- /dev/null +++ b/mlruns/0/b19dc8b40ada4fdea65402cf2f8a4c39/tags/mlflow.user @@ -0,0 +1 @@ +tomasz \ No newline at end of file diff --git a/train.py b/train.py index 67d14d7..ed4dd25 100755 --- a/train.py +++ b/train.py @@ -40,8 +40,10 @@ def prepare_model(epoch_param, batch_param, _run): X_train = vgsales_train.drop(['Rank','Name','Platform','Year','Genre','Publisher'],axis = 1) y_train = vgsales_train[['Nintendo']] + y_train = keras.utils.to_categorical(y_train) X_test = vgsales_test.drop(['Rank','Name','Platform','Year','Genre','Publisher'],axis = 1) y_test = vgsales_test[['Nintendo']] + y_test = keras.utils.to_categorical(y_test) print(X_train.shape[1]) @@ -51,7 +53,7 @@ def prepare_model(epoch_param, batch_param, _run): early_stop = EarlyStopping(monitor="val_loss", mode="min", verbose=1, patience=10) - model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy']) + model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy']) epochs = int(sys.argv[1]) batch_size = int(sys.argv[2]) diff --git a/vgsales-mlflow.py b/vgsales-mlflow.py index 7ef1f2a..81716fb 100644 --- a/vgsales-mlflow.py +++ b/vgsales-mlflow.py @@ -35,7 +35,7 @@ def my_main(epochs, batch_size): early_stop = EarlyStopping(monitor="val_loss", mode="min", verbose=1, patience=10) - model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy']) + model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy']) model.fit(X_train, y_train, epochs=epochs, batch_size=batch_size, validation_data=(X_test, y_test))