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)
|
X_train = vgsales_train.drop(['Rank','Name','Platform','Year','Genre','Publisher'],axis = 1)
|
||||||
y_train = vgsales_train[['Nintendo']]
|
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)
|
X_test = vgsales_test.drop(['Rank','Name','Platform','Year','Genre','Publisher'],axis = 1)
|
||||||
y_test = vgsales_test[['Nintendo']]
|
y_test = vgsales_test[['Nintendo']]
|
||||||
|
y_test = keras.utils.to_categorical(y_test)
|
||||||
|
|
||||||
print(X_train.shape[1])
|
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)
|
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])
|
epochs = int(sys.argv[1])
|
||||||
batch_size = int(sys.argv[2])
|
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)
|
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))
|
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