Skip to content

Tag: scikit-learn

K-Fold cross validation for Lasso and Ridge models

I’m working with the Boston housing dataset from sklearn.datasets and have run ridge and lasso regressions on my data (post train/test split). I’m now trying to perform k-fold cross validation to find the optimal penalty parameters, and have written the code below. What can I do to resolve this is…

How to change plot legends with roc_auc_score?

I’m plotting ROC curve with plot_roc_curve of scikit-learn, that plot legends are printed automatically. Is there a way to change them? Answer You can pretty much add anything you like to the plot object that is produced through plot_roc_curve. For instance, you can do something like this: This will ret…