diff --git a/requirements.txt b/requirements.txt index 64761ba..b2d8923 100644 --- a/requirements.txt +++ b/requirements.txt @@ -13,7 +13,7 @@ joblib==1.0.1 MarkupSafe==2.0.1 numpy==1.21.1 packaging==21.3 -pandas==1.3.1 +pandas==1.4.2 pluggy==1.0.0 py==1.11.0 pyparsing==3.0.6 @@ -22,7 +22,7 @@ python-dateutil==2.8.2 pytz==2021.1 requests==2.26.0 scikit-learn==0.24.2 -scipy==1.7.1 +scipy==1.8.0 six==1.16.0 threadpoolctl==2.2.0 toml==0.10.2 diff --git a/src/predict.py b/src/predict.py index 4a85a35..1a36f51 100644 --- a/src/predict.py +++ b/src/predict.py @@ -5,8 +5,7 @@ file_to_open = open("data/models/baummethoden_lr.pickle", 'rb') trained_model = pickle.load(file_to_open) file_to_open.close() - +i =0 # load data that we want predictions for prediction_data = pd.read_csv('data/prediction-data.csv', sep=";") - print(trained_model.predict(prediction_data))