Spaces:

ThirdEyeData
/

Component_Repair_Time_Prediction

Sleeping

saritha5 commited on Apr 1, 2023

Commit

f4afd35

1 Parent(s): aa298e4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -141,8 +141,8 @@ def prediction(df):
   #X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.1, random_state=42)
   #print(X_train.shape)
   #print(X_test.shape)
-  X_test_encoded = label_encoding(df)
-  X_test_df = preprocess_dataset(X_test_encoded)
   x_model = pickle.load(open('repair_time_model.pkl','rb'))
   pred = x_model.predict(X_test_df)
   #X_test['Actual_time_to_repair'] = y_test

   #X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.1, random_state=42)
   #print(X_train.shape)
   #print(X_test.shape)
+  #X_test_encoded = label_encoding(df)
+  X_test_df = preprocess_dataset(df)
   x_model = pickle.load(open('repair_time_model.pkl','rb'))
   pred = x_model.predict(X_test_df)
   #X_test['Actual_time_to_repair'] = y_test