From fc59ee8b36c31304ecb7ab5ab5c94429d9a65825 Mon Sep 17 00:00:00 2001 From: Kacper Dudzic Date: Sun, 24 Apr 2022 22:57:32 +0200 Subject: [PATCH] Update 'simple_regression.py' --- simple_regression.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/simple_regression.py b/simple_regression.py index a5824e5..7b2d471 100644 --- a/simple_regression.py +++ b/simple_regression.py @@ -48,7 +48,7 @@ history = model.fit( y_pred = model.predict(test_piece_counts) # Zapis predykcji do pliku -results = pd.DataFrame({"test_set_piece_count": test_piece_counts.numpy().tolist(), "predicted_price": [a[0] for a in y_pred.tolist()]}) +results = pd.DataFrame({"test_set_piece_count": test_piece_counts.tolist(), "predicted_price": [a[0] for a in y_pred.tolist()]}) results.to_csv(r'lego_reg_results.csv', index=False, header=True) # Zapis modelu do pliku