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')