Spaces:

Srastog
/

Machine-Downtime-Prediction-API

Running

Srastog commited on Jan 22

Commit

97ce236

verified ·

1 Parent(s): affac97

Update app/modelling.py

Files changed (1) hide show

app/modelling.py CHANGED Viewed

@@ -10,8 +10,7 @@ from sklearn.metrics import f1_score
 import argparse
 import joblib
-def train(dataset_pth):
-        df=pd.read_csv(dataset_pth)
         features=["Torque(Nm)","Hydraulic_Pressure(bar)","Cutting(kN)","Coolant_Pressure(bar)","Spindle_Speed(RPM)","Coolant_Temperature","Downtime"]
         df=df[features]

 import argparse
 import joblib
+def train(df):
         features=["Torque(Nm)","Hydraulic_Pressure(bar)","Cutting(kN)","Coolant_Pressure(bar)","Spindle_Speed(RPM)","Coolant_Temperature","Downtime"]
         df=df[features]