snajmark commited on
Commit
25d5b5a
·
1 Parent(s): f874cee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -413,6 +413,10 @@ def predict_inverse(antimicrobial_activity_target, request: gr.Request):
413
  optimized_x = pd.DataFrame.from_dict(best_params)
414
  optimized_x[numerical_columns] = minmax_scaler_inputs.inverse_transform(optimized_x[numerical_columns])
415
 
 
 
 
 
416
  return optimized_x
417
 
418
 
 
413
  optimized_x = pd.DataFrame.from_dict(best_params)
414
  optimized_x[numerical_columns] = minmax_scaler_inputs.inverse_transform(optimized_x[numerical_columns])
415
 
416
+ for column in optimized_x.columns:
417
+ if column in one_hot_mapping:
418
+ optimized_x.loc[0, column] = one_hot_mapping[column][int(optimized_x.loc[0, column])]
419
+
420
  return optimized_x
421
 
422