Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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[
|
| 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):
|