Arafath10 commited on
Commit
8d31b67
·
verified ·
1 Parent(s): 6cb03a2

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +7 -1
main.py CHANGED
@@ -214,12 +214,17 @@ def predict(customer_name: str,
214
  def safe_transform(encoder, column):
215
  classes = encoder.classes_
216
  return [encoder.transform([x])[0] if x in classes else -1 for x in column]
 
 
 
 
 
217
 
218
  # Convert input data to DataFrame
219
  input_data = {
220
  'customer_name': customer_name,
221
  'customer_address': customer_address,
222
- 'customer_phone': customer_phone,
223
  'customer_email': customer_email,
224
  'cod': float(cod),
225
  'weight': float(weight),
@@ -228,6 +233,7 @@ def predict(customer_name: str,
228
  }
229
  input_df = pd.DataFrame([input_data])
230
 
 
231
  # Encode categorical variables using the same encoders used during training
232
  for col in input_df.columns:
233
  if col in encoders:
 
214
  def safe_transform(encoder, column):
215
  classes = encoder.classes_
216
  return [encoder.transform([x])[0] if x in classes else -1 for x in column]
217
+ # Function to handle unseen labels during encoding
218
+ def safe_transform(encoder, column):
219
+ classes = encoder.classes_
220
+ return [encoder.transform([x])[0] if x in classes else -1 for x in column]
221
+
222
 
223
  # Convert input data to DataFrame
224
  input_data = {
225
  'customer_name': customer_name,
226
  'customer_address': customer_address,
227
+ 'customer_phone': int(customer_phone),
228
  'customer_email': customer_email,
229
  'cod': float(cod),
230
  'weight': float(weight),
 
233
  }
234
  input_df = pd.DataFrame([input_data])
235
 
236
+
237
  # Encode categorical variables using the same encoders used during training
238
  for col in input_df.columns:
239
  if col in encoders: