Zaktualizuj 'ml_pytorch_mlflow.py'

This commit is contained in:
Sebastian Wałęsa 2022-05-15 22:52:20 +02:00
parent 2eed47884d
commit 84ba2ea32a

View File

@ -136,7 +136,7 @@ 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 = np.append(inputs, input_) inputs = inputs.append(input_)
inputs = pd.DataFrame(inputs, dtype=np.float64) inputs = pd.DataFrame(inputs, dtype=np.float64)
inputs = inputs.to_numpy() inputs = inputs.to_numpy()