From 77ff5d4a7ee89378ea232fe7d4503b0fc7173a17 Mon Sep 17 00:00:00 2001 From: Kacper Dudzic Date: Sun, 24 Apr 2022 23:05:58 +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 aee6033..fa31f96 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.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].round(2) for a in y_pred.tolist()]}) results.to_csv('lego_reg_results.csv', index=False, header=True) # Zapis modelu do pliku