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 |
|