sklearn hackerrank