diff --git a/ml_pytorch_mlflow.py b/ml_pytorch_mlflow.py index ab1bf37..f45325f 100644 --- a/ml_pytorch_mlflow.py +++ b/ml_pytorch_mlflow.py @@ -141,9 +141,9 @@ def my_main(epochs): # inputss, targetss = val_ds - inputss = pd.DataFrame(inputss, dtype=np.float64) + inputsss = pd.DataFrame(inputss, dtype=np.float64) # inputss = inputss.to_numpy() - inputsss = inputss.values + # inputsss = inputss.values MSE = mean_squared_error(expected, predicted) MAE = mean_absolute_error(expected, predicted)