adityaprakash17
commited on
Commit
•
e14c248
1
Parent(s):
38cc412
Update app.py
Browse files
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):
|