Spaces:
Sleeping
Sleeping
Commit ·
5948f46
1
Parent(s): 08a73ed
null
Browse files
app.py
CHANGED
|
@@ -19,12 +19,18 @@ def normalize(value, cast=None):
|
|
| 19 |
|
| 20 |
|
| 21 |
def predict_job(job_data):
|
| 22 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 23 |
|
| 24 |
# Predict
|
| 25 |
-
probability = model.predict_proba(df)[0][1]
|
| 26 |
prediction = "HIRED" if probability > 0.5 else "NOT_HIRED"
|
| 27 |
-
|
| 28 |
return {
|
| 29 |
'prediction': prediction,
|
| 30 |
'hire_probability': f"{probability*100:.1f}",
|
|
|
|
| 19 |
|
| 20 |
|
| 21 |
def predict_job(job_data):
|
| 22 |
+
import numpy as np
|
| 23 |
+
|
| 24 |
+
# Replace None with np.nan to ensure numeric dtype
|
| 25 |
+
clean_data = {k: (np.nan if v is None else v) for k, v in job_data.items()}
|
| 26 |
+
df = pd.DataFrame([clean_data])
|
| 27 |
+
|
| 28 |
+
# Force all columns to float64 to match training data
|
| 29 |
+
df = df.astype('float64')
|
| 30 |
|
| 31 |
# Predict
|
| 32 |
+
probability = model.predict_proba(df)[0][1]
|
| 33 |
prediction = "HIRED" if probability > 0.5 else "NOT_HIRED"
|
|
|
|
| 34 |
return {
|
| 35 |
'prediction': prediction,
|
| 36 |
'hire_probability': f"{probability*100:.1f}",
|