Olá pessoal, visando a contribuição com a comunidade, resolvi procurar na internet soluções pra divisão do dataset em 3 (pra treino teste e validação) e encontrei essa lib:
https://pypi.org/project/fast-ml/
Model Development
from fast_ml.model_development import train_valid_test_split
X_train, y_train, X_valid, y_valid, X_test, y_test = train_valid_test_split(df, target = target,
train_size=0.8, valid_size=0.1, test_size=0.1)
Espero ter ajudado!