mlflow
Some checks failed
s444501-training/pipeline/head There was a failure building this commit

This commit is contained in:
s444501 2022-05-13 02:03:19 +02:00
parent de92ae7ae9
commit bd9d4523af

View File

@ -5,11 +5,13 @@ import mlflow
import torch
import torch.nn as nn
import torch.nn.functional as F
from mlflow.experiments import restore_experiment
from sklearn.preprocessing import LabelEncoder
import pandas as pd
# MLFlow
mlflow.set_tracking_uri("http://172.17.0.1:5000")
restore_experiment(15)
mlflow.set_experiment("s444501")