Update app/modelling.py

app/modelling.py

@@ -33,16 +33,10 @@ def train(dataset):
     model.fit(X_train,y_train)
     predict=model.predict(X_test)
 
-    cwd=os.getcwd()
-    transform_pth=os.path.join(cwd,"app","transform.pkl")
-    encoder_pth=os.path.join(cwd,"app","encoder.pkl")
-    model_pth=os.path.join(cwd,"app","model.pkl")
-
-    joblib.dump(transform,transform_pth)
-    joblib.dump(encoder,encoder_pth)
-    joblib.dump(model,model_pth)
-
-    return {"Accuracy":f"{accuracy_score(y_test,predict):4f}",
+    return {"model":model,
+            "encoder":encoder,
+            "transform":transform,
+            "Accuracy":f"{accuracy_score(y_test,predict):4f}",
             "F1_Score":f"{f1_score(y_test,predict):4f}"}