Spaces:
Sleeping
Sleeping
Update main.py
Browse files
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:
|