tzietkiewicz/ium-helloworld
1
0
Fork 1
Code Issues Pull Requests Projects Releases Wiki Activity

Change name of the experiment
Some checks failed
gitea-test/ium-helloworld/pipeline/head There was a failure building this commit
Details

Browse Source 
to avoid colision with deleted one
This commit is contained in:
Tomasz 2023-04-20 10:55:18 +02:00
parent af2c3ae5d4
commit 253e44450f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -21,7 +21,7 @@ logging.basicConfig(level=logging.WARN)
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
mlflow.set_tracking_uri("http://172.17.0.1:5000") mlflow.set_tracking_uri("http://172.17.0.1:5000")
mlflow.set_experiment("s123456-new") mlflow.set_experiment("s123456-2023")
def eval_metrics(actual, pred): def eval_metrics(actual, pred):
rmse = np.sqrt(mean_squared_error(actual, pred)) rmse = np.sqrt(mean_squared_error(actual, pred))