fix
This commit is contained in:
parent
b5774c67b7
commit
9050bcb8e7
15
mlruns/0/156c624fa6ff468887bf9543a37ce8b3/meta.yaml
Normal file
15
mlruns/0/156c624fa6ff468887bf9543a37ce8b3/meta.yaml
Normal 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
|
@ -0,0 +1 @@
|
||||
b5774c67b74e143c2e0d3a990180a7305140b234
|
@ -0,0 +1 @@
|
||||
/home/tomasz/IUM/ium_434695/vgsales-mlflow.py
|
@ -0,0 +1 @@
|
||||
LOCAL
|
@ -0,0 +1 @@
|
||||
tomasz
|
15
mlruns/0/4be4d7d988dc477e8ec483ca0ec5bb30/meta.yaml
Normal file
15
mlruns/0/4be4d7d988dc477e8ec483ca0ec5bb30/meta.yaml
Normal 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
|
@ -0,0 +1 @@
|
||||
b5774c67b74e143c2e0d3a990180a7305140b234
|
@ -0,0 +1 @@
|
||||
/home/tomasz/IUM/ium_434695/vgsales-mlflow.py
|
@ -0,0 +1 @@
|
||||
LOCAL
|
@ -0,0 +1 @@
|
||||
tomasz
|
15
mlruns/0/b19dc8b40ada4fdea65402cf2f8a4c39/meta.yaml
Normal file
15
mlruns/0/b19dc8b40ada4fdea65402cf2f8a4c39/meta.yaml
Normal 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
|
@ -0,0 +1 @@
|
||||
b5774c67b74e143c2e0d3a990180a7305140b234
|
@ -0,0 +1 @@
|
||||
/home/tomasz/IUM/ium_434695/vgsales-mlflow.py
|
@ -0,0 +1 @@
|
||||
LOCAL
|
@ -0,0 +1 @@
|
||||
tomasz
|
4
train.py
4
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])
|
||||
|
@ -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))
|
||||
|
Loading…
Reference in New Issue
Block a user