Train split sklearn (18/36)

Train split sklearn