s434695/ium_434695
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix

Browse Source
This commit is contained in:
s434695 2021-05-24 00:30:08 +02:00
parent a024e32143
commit c212ffdb9c
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vgsales-mlflow.py
View File

@ -14,10 +14,12 @@ from tensorflow.keras.models import Model
from tensorflow.keras.callbacks import EarlyStopping from tensorflow.keras.callbacks import EarlyStopping
from tensorflow.keras.models import Sequential from tensorflow.keras.models import Sequential
import mlflow import mlflow
from mlflow.tracking import MlflowClient
mlflow.set_tracking_uri("http://172.17.0.1:5000") mlflow.set_tracking_uri("http://172.17.0.1:5000")
mlflow.set_experiment("s434749") mlflow.set_experiment("s434749")
client = MlflowClient()
def my_main(epochs, batch_size): def my_main(epochs, batch_size):