added rmse to sacred
This commit is contained in:
parent
6596e3373e
commit
9b2764a4c0
@ -23,7 +23,7 @@ def my_config():
|
|||||||
batch_size = 16
|
batch_size = 16
|
||||||
|
|
||||||
@ex.capture
|
@ex.capture
|
||||||
def prepare_model(epochs, batch_size):
|
def prepare_model(epochs, batch_size, _run):
|
||||||
# odczytanie danych z plików
|
# odczytanie danych z plików
|
||||||
avocado_train = pd.read_csv('avocado_train.csv')
|
avocado_train = pd.read_csv('avocado_train.csv')
|
||||||
avocado_test = pd.read_csv('avocado_test.csv')
|
avocado_test = pd.read_csv('avocado_test.csv')
|
||||||
@ -55,6 +55,7 @@ def prepare_model(epochs, batch_size):
|
|||||||
|
|
||||||
# ewaluacja
|
# ewaluacja
|
||||||
rmse = mean_squared_error(y_test, prediction)
|
rmse = mean_squared_error(y_test, prediction)
|
||||||
|
_run.log_scalar("rmse", rmse)
|
||||||
|
|
||||||
# zapisanie modelu
|
# zapisanie modelu
|
||||||
model.save('avocado_model.h5')
|
model.save('avocado_model.h5')
|
||||||
|
Loading…
Reference in New Issue
Block a user