Zaktualizuj 'ml_pytorch_mlflow.py'

This commit is contained in:
Sebastian Wałęsa 2022-05-15 23:10:52 +02:00
parent 383e2ed7b8
commit ccff4dcfad

View File

@ -142,7 +142,7 @@ def my_main(epochs):
# inputss, targetss = val_ds
inputss = pd.DataFrame(inputss, dtype=np.float64)
# inputss = inputss.to_numpy()
inputss = inputss.to_numpy()
MSE = mean_squared_error(expected, predicted)
MAE = mean_absolute_error(expected, predicted)