Zaktualizuj 'ml_pytorch_mlflow.py'
This commit is contained in:
parent
4eaf748f2d
commit
6f47a1390c
@ -135,8 +135,9 @@ def my_main(epochs):
|
||||
input_, target = val_ds[i]
|
||||
expected.append(float(target))
|
||||
predicted.append(float(prediction(input_, model)))
|
||||
input_ = pd.DataFrame(input_)
|
||||
inputs.append(input_)
|
||||
|
||||
inputs = pd.DataFrame(inputs)
|
||||
|
||||
MSE = mean_squared_error(expected, predicted)
|
||||
MAE = mean_absolute_error(expected, predicted)
|
||||
|
Loading…
Reference in New Issue
Block a user