Solucionado
Saudações, Nos cursos anteriores, utilizamos o argumento stratify na função train_test_split para melhorar a qualidade da separação, porém utilizamos eles para classificações booleanas. No caso de regressões, também é interessante utilizar este argumento?
Oi, Alison, tudo bem?
Na regressão, o argumento stratify não é utilizado, pois ele é usado para garantir que a proporção das classes seja mantida nas amostras de treino e teste em problemas de classificação. Na regressão, o objetivo é prever um valor contínuo, e não fazer uma separação binária.
Espero ter ajudado. Caso tenha dúvidas, conte com o fórum. Abraços!