text_html_output_1.png
import matplotlib.pyplot as plt

import pandas as pd

import seaborn as sns

train = pd.read_csv('/kaggle/input/playground-series-s3e14/train.csv')
test = pd.read_csv('/kaggle/input/playground-series-s3e14/test.csv')

features = ['clonesize', 'honeybee', 'bumbles', 'andrena', 'osmia', 'AverageRainingDays', 'fruitset', 'fruitmass', 'seeds']
X = train[features]
y = train['yield']

test_features = test[features]
test['yield'] = grid_search.predict(test_features)
test[['id', 'yield']].to_csv('submission.csv', index=False)
display(test[['id', 'yield']])