fortnite_knn_model_selection.py

r2df = pd.DataFrame(r2_scores, columns=["k", "train_data_score", "test_data_score"])
f, ax = plt.subplots(1, 1, figsize=(30,5))
ax.scatter(r2df.k, r2df.train_data_score, label='Train data score')                
ax.scatter(r2df.k, r2df.test_data_score, color='r', label='Test data score')       
plt.xticks(r2df.k)
plt.legend()
plt.xlabel('k values')
plt.ylabel(r'$R^2$')
plt.grid(False)