From 59f8252fe1d514d945a21edf9bb19ae0a307b8cf Mon Sep 17 00:00:00 2001 From: Kacper Dudzic Date: Sun, 24 Apr 2022 23:03:46 +0200 Subject: [PATCH] Update 'simple_regression.py' --- simple_regression.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/simple_regression.py b/simple_regression.py index 7b2d471..aee6033 100644 --- a/simple_regression.py +++ b/simple_regression.py @@ -48,8 +48,8 @@ history = model.fit( y_pred = model.predict(test_piece_counts) # Zapis predykcji do pliku -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) +results = pd.DataFrame({'test_set_piece_count': test_piece_counts.tolist(), 'predicted_price': [a[0] for a in y_pred.tolist()]}) +results.to_csv('lego_reg_results.csv', index=False, header=True) # Zapis modelu do pliku model.save('lego_reg_model')