의사결정나무트리1 타이타닉승객들의 생존률 예측하기_모델 생성 및 학습 저번 시간에는 모델 학습에 필요한 데이터들을 전처리하는 과정을 거쳐 보았다. 이번 시간에는 이 데이터들을 바탕으로 모델을 생성하고 학습을 해보도록 하자. Feature Engineering & Initial Modeling 1 2 X_train = train[['Sex','Pclass']] y_train = train[['Survived']] cs 이번예제에서는 성별과 객실 정보를 사용하여 학습을 시켜보도록 하자. x_train에 'Sex' 와 'Pclass'를 불러오고 예측혜야하는 값인 'Survived'를 y_train에 불러들이도록 한다. 1 X_test = test[['Sex','Pclass']] cs X_test에는 X_train과 마찬가지로 'Sex' 와 'Pclass'를 불러들인다. skle.. 2021. 8. 7. 이전 1 다음