This commit is contained in:
Wojciech Lidwin 2023-05-12 04:39:40 +02:00
parent 19b8ac4d4a
commit cdf8523c94

View File

@ -53,7 +53,7 @@ def my_config():
def prepare_message(epochs, lr, validation_split):
return "{0} {1} {2}!".format(epochs, lr, validation_split)
@ex.main
@ex.automain
def my_main(epochs, lr, validation_split, _run):
train = pd.read_csv('baltimore_train.csv')