Senasu commited on
Commit
6d7f375
·
verified ·
1 Parent(s): 8ebef03

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ preprocessor = ColumnTransformer(
22
 
23
  # Pipeline for preprocessing and model prediction
24
  pipeline = Pipeline(steps=[('preprocessor', preprocessor), ('regressor', model)])
25
- pipeline.fit(df[['R&D Spend', 'Administration', 'Marketing Spend', 'State']], df[['Profit']])
26
 
27
  # Prediction function
28
  def price_pred(rd, administration, marketing, state):
 
22
 
23
  # Pipeline for preprocessing and model prediction
24
  pipeline = Pipeline(steps=[('preprocessor', preprocessor), ('regressor', model)])
25
+ pipeline.fit(df[['R&D Spend', 'Administration', 'Marketing Spend', 'State']], df['Profit'])
26
 
27
  # Prediction function
28
  def price_pred(rd, administration, marketing, state):