saifhmb
commited on
Update prediction formula to model.predict(sc.transform([[Age, EstimatedSalary]]))
Browse files
app.py
CHANGED
|
@@ -62,7 +62,7 @@ def welcome():
|
|
| 62 |
|
| 63 |
# defining the function which will make the prediction using the data which the user inputs
|
| 64 |
def prediction(Age, EstimatedSalary):
|
| 65 |
-
prediction = model.predict
|
| 66 |
print(prediction)
|
| 67 |
return prediction
|
| 68 |
|
|
|
|
| 62 |
|
| 63 |
# defining the function which will make the prediction using the data which the user inputs
|
| 64 |
def prediction(Age, EstimatedSalary):
|
| 65 |
+
prediction = model.predict(sc.transform([[Age, EstimatedSalary]]))
|
| 66 |
print(prediction)
|
| 67 |
return prediction
|
| 68 |
|