Zaktualizuj 'train-mlflow.py'

2022-05-15 15:35:02 +02:00 · 2022-05-15 15:35:02 +02:00 · 357d3e7052
commit 357d3e7052
parent 777b126c73
1 changed files with 2 additions and 2 deletions
--- a/train-mlflow.py
+++ b/train-mlflow.py
@ -91,8 +91,8 @@ def my_main(epochs):
        with open ("output.txt",'a+') as f:
            if (epochs%100==0):
                f.write(f'epoch:{epochs+1},loss = {loss.item():.4f}')
-        x = pd.DataFrame(X_training)
-        y = pd.DataFrame(y_predicted_train)
+        x = X_training.detach()
+        y = y_predicted_train.detach()

        with torch.no_grad():
            y_predicted = model(X_testing)