HOLYBOY commited on
Commit
09071a1
1 Parent(s): 07ecaec

New Update

Browse files
Files changed (1) hide show
  1. app.py +17 -0
app.py CHANGED
@@ -16,4 +16,21 @@ scaler = joblib.load('https://drive.google.com/file/d/1qdh31R8f7KzP3UZY9rZkMUrtx
16
  lr_model = joblib.load('https://drive.google.com/file/d/1qdh31R8f7KzP3UZY9rZkMUrtxDbX69sr/view?usp=drive_link/lr_smote_model.joblib')
17
 
18
 
 
 
19
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
16
  lr_model = joblib.load('https://drive.google.com/file/d/1qdh31R8f7KzP3UZY9rZkMUrtxDbX69sr/view?usp=drive_link/lr_smote_model.joblib')
17
 
18
 
19
+ def preprocess_input(input_data):
20
+ input_df = pd.DataFrame(input_data, index=[0])
21
 
22
+ cat_columns = [col for col in input_df.columns if input_df[col].dtype == 'object']
23
+ num_columns = [col for col in input_df.columns if input_df[col].dtype != 'object']
24
+
25
+ input_df_imputed_cat = cat_imputer.transform(input_df[cat_columns])
26
+ input_df_imputed_num = num_imputer.transform(input_df[num_columns])
27
+
28
+ input_encoded_df = pd.DataFrame(encoder.transform(input_df_imputed_cat).toarray(),
29
+ columns=encoder.get_feature_names_out(cat_columns))
30
+
31
+ input_df_scaled = scaler.transform(input_df_imputed_num)
32
+ input_scaled_df = pd.DataFrame(input_df_scaled, columns=num_columns)
33
+ final_df = pd.concat([input_encoded_df, input_scaled_df], axis=1)
34
+ final_df = final_df.reindex(columns=original_feature_names, fill_value=0)
35
+
36
+ return final_df