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