update dllib-mlflow.py
Some checks failed
s444356-evaluation/pipeline/head This commit looks good
s444356-training/pipeline/head There was a failure building this commit

This commit is contained in:
Maciej Czajka 2022-05-09 22:08:18 +02:00
parent 2cb59732d1
commit b0c7aba70a

View File

@ -2,6 +2,7 @@ import numpy as np
import sys
import os
import torch
Import mlflow
import pandas as pd
from torch import nn
from torch.autograd import Variable