yusufenes commited on
Commit
69cc580
·
verified ·
1 Parent(s): 4edc7e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -53,13 +53,9 @@ def model_fit(X_train,y_train):
53
  hgb = HistGradientBoostingRegressor()
54
  hgb.fit(X_train,y_train)
55
  return hgb
56
-
57
  model = model_fit(X_train,y_train)
58
 
59
-
60
-
61
- df = pd.get_dummies(df,drop_first=True)
62
-
63
  import streamlit as st
64
 
65
  def price(companyName,modelName,modelYear,locaiton,mileage,engineType,engineCapacity,color,assembly,bodyType,transmissionType,registrationStatus):
 
53
  hgb = HistGradientBoostingRegressor()
54
  hgb.fit(X_train,y_train)
55
  return hgb
56
+ df,X_train,X_test,y_train,y_test = final_df(df)
57
  model = model_fit(X_train,y_train)
58
 
 
 
 
 
59
  import streamlit as st
60
 
61
  def price(companyName,modelName,modelYear,locaiton,mileage,engineType,engineCapacity,color,assembly,bodyType,transmissionType,registrationStatus):