Skip to content

Commit

Permalink
Update data_training.py
Browse files Browse the repository at this point in the history
  • Loading branch information
yaswanthsai002 authored Sep 28, 2023
1 parent f8e44dc commit 4937420
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions app/data_training.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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)

0 comments on commit 4937420

Please sign in to comment.