This commit is contained in:
s434695 2021-05-23 23:14:15 +02:00
parent b5774c67b7
commit 9050bcb8e7
17 changed files with 61 additions and 2 deletions

View File

@ -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

View File

@ -0,0 +1 @@
b5774c67b74e143c2e0d3a990180a7305140b234

View File

@ -0,0 +1 @@
/home/tomasz/IUM/ium_434695/vgsales-mlflow.py

View File

@ -0,0 +1 @@
LOCAL

View File

@ -0,0 +1 @@
tomasz

View File

@ -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

View File

@ -0,0 +1 @@
b5774c67b74e143c2e0d3a990180a7305140b234

View File

@ -0,0 +1 @@
/home/tomasz/IUM/ium_434695/vgsales-mlflow.py

View File

@ -0,0 +1 @@
LOCAL

View File

@ -0,0 +1 @@
tomasz

View File

@ -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

View File

@ -0,0 +1 @@
b5774c67b74e143c2e0d3a990180a7305140b234

View File

@ -0,0 +1 @@
/home/tomasz/IUM/ium_434695/vgsales-mlflow.py

View File

@ -0,0 +1 @@
LOCAL

View File

@ -0,0 +1 @@
tomasz

View File

@ -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])

View File

@ -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))