인프런 커뮤니티 질문&답변

안녕하세요.

아래 코드 중 predict() 과정 부분에서 궁금중이 생겨서 질문 드립니다.

아래 코드를 보면 dt_clf에 DecisionTreeClassifier 객체를 생성후 fit()을 실행하는 코드가 있는데

저는 이부분도 학습된 데이터를 객체에 생성해서

그 객체 명으로  predict()를 실행할 줄 알았는데

ex) dt_clf_fit = dt_clf.fit(X_train, y_train)

-> pred = dt_clf_fit.predict(X_test)

책 설명을 보니 fit()을 실행 한 순간

dt_clf가 학습된 데이터가 적용이 되고 이 객체를 predict()를 실행하는거 같은데 제가 이해하고 있는 부분이 맞는지 궁금합니다.

 

# DecisionTreeClassifier 객체 생성 

dt_clf = DecisionTreeClassifier(random_state=11)

# 학습 수행 

dt_clf.fit(X_train, y_train)

# 학습이 완료된 DecisionTreeClassifier 객체에서 테스트 데이터 세트로 예측 수행. 

pred = dt_clf.predict(X_test)