Update main.py

main.py

@@ -160,7 +160,7 @@ def evaluate_model(y_test, y_pred):
 
 
 if __name__ == "__main__":
-    df_train = pd.read_csv(str(Path(__file__).parents[1] / 'data/train.csv'))  # Load Data
+    df_train = pd.read_csv(str(Path(__file__).parents[1] / 'train.csv'))  # Load Data
     cleaning_steps(df_train)  # Perform Cleaning
     extract_label_value(df_train) #Extract Label Value
     perform_feature_engineering(df_train)  # Perform feature engineering
@@ -182,5 +182,5 @@ if __name__ == "__main__":
     evaluate_model(y_test, y_pred)
 
     # Save Model
-    with open(str(Path(__file__).parents[1] / 'code/model.pickle'), 'wb') as f:
+    with open(str(Path(__file__).parents[1] / 'model.pickle'), 'wb') as f:
         pickle.dump((model, label_encoders, scaler), f)