Put the sklearn bit