saifhmb commited on
Commit
d3fd480
·
unverified ·
1 Parent(s): d6a6f25

Update prediction formula to model.predict(sc.transform([[Age, EstimatedSalary]]))

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.sc.transform([[Age, EstimatedSalary]])
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