adityaprakash17 commited on
Commit
e14c248
1 Parent(s): 38cc412

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -18,13 +18,11 @@ x_train, x_test, y_train, y_test = train_test_split(df.drop('charges',axis=1), d
18
  #dtree = DecisionTreeRegressor()
19
  #model training
20
  #dtree.fit(x_train,y_train)
21
- x_train=x_train.values
22
- y_train=y_train.values
23
 
24
  from sklearn.ensemble import RandomForestRegressor
25
  rf = RandomForestRegressor()
26
  #model training
27
- rf.fit(x_train,y_train)
28
 
29
 
30
  def func(a, b, c, d, e, f):
 
18
  #dtree = DecisionTreeRegressor()
19
  #model training
20
  #dtree.fit(x_train,y_train)
 
 
21
 
22
  from sklearn.ensemble import RandomForestRegressor
23
  rf = RandomForestRegressor()
24
  #model training
25
+ rf.fit(x_train.values,y_train.values)
26
 
27
 
28
  def func(a, b, c, d, e, f):