diff --git a/app/data_training.py b/app/data_training.py index 2101670..4925822 100644 --- a/app/data_training.py +++ b/app/data_training.py @@ -5,7 +5,7 @@ from sklearn.model_selection import train_test_split, cross_val_score # Load the data -df = pd.read_csv("Asana.csv") +df = pd.read_csv("app/Asana.csv") # Split the data into input features (X) and target variable (y) X = df.iloc[:,:-1].values @@ -36,5 +36,5 @@ print("Random Forest Classifier : ", round(accuracy, 2)) # Save the model and label encoder to a file -with open('model.pkl', 'wb') as f: +with open('app/model.pkl', 'wb') as f: pickle.dump((rfr, le), f)