Olá! Quando for dividir os dados entre teste e treino posso usar biblioteca do skitlearn que faz a separação?
from sklearn.model_selection import train_test_split
Você está vendo a versão anterior da nova experiência da Alura que estamos preparando para você. Em breve, ela ganha uma identidade visual novinha totalmente pensada em potencializar seus estudos!
Solucionado
Olá! Quando for dividir os dados entre teste e treino posso usar biblioteca do skitlearn que faz a separação?
from sklearn.model_selection import train_test_split
Oi Leticia,
Pode sim. Porém, o padrão da train_test_split do sklearn é fazer uma divisão aleatória para o treino e para o teste. Assim teríamos parte do que usamos para o treino no curso sendo usada para o teste e vice-versa. Para fazer um train_test_split sequencial como fizemos no curso recomendo usar a train_test_split da biblioteca pmdarima. Essa train_test_split é uma versão modificada da versão do sklearn para trabalhar de forma sequencial.
Nas aulas seguintes do curso vamos instalar essa biblioteca e acho que vai ficar mais fácil de entender. Mas aqui segue um exemplo de uso. Copiado da documentação da pmdarima.
import pmdarima as pm
from pmdarima.model_selection import train_test_split
y = pm.datasets.load_sunspots()
y_train, y_test = train_test_split(y, test_size=50)
y_test.shape
(50,)