Spaces:

ThirdEyeData
/

Component_Repair_Time_Prediction

Sleeping

saritha5 commited on Apr 1, 2023

Commit

b2cfd57

1 Parent(s): f4afd35

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -142,9 +142,9 @@ def prediction(df):
   #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
   #X_test['Predicted_time_to_repair'] = pred
   #X_test.to_csv(r'/content/drive/MyDrive/Colab Notebooks/HAL/repair_time_prediction_results.csv')

   #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(df)
   #X_test['Actual_time_to_repair'] = y_test
   #X_test['Predicted_time_to_repair'] = pred
   #X_test.to_csv(r'/content/drive/MyDrive/Colab Notebooks/HAL/repair_time_prediction_results.csv')