Zaktualizuj 'ml_pytorch_mlflow.py'

This commit is contained in:
Sebastian Wałęsa 2022-05-15 23:09:33 +02:00
parent cc3a00146c
commit 383e2ed7b8

View File

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