Pergunta de entrevista da empresa Sony

Implement a dataset splitting function in Python or JS, then once that was concluded we moved on to make a stratified version.