빅분기-작업형-유형2-의사결정나무-분류-Iris

macbook-type2-iris

—————

 

유형2 

지도-분류: 의사결정나무, KNN, SVM, Logistic Regress, RandomForest, NeuralNetwork

지도-회귀: 선형회귀분석, 다중회귀분석, 의사결정나무 회귀

비지도: 군집Clustering, 연관분석AssocialtionAnalysis, 인공신경망NN

 

 

Type2: DecisionTreeClassifier(의사결정나무-분류)-IRIS 

 

붓꽃 품종을 분류(예측)하는 분석 전체 코드

 

import numpy as np

import pandas as pd

import sklearn

 

from sklearn.tree import DecisionTreeClassifier

from sklearn.model_selection import train_test_split

 

df = pd.read_csv('/content/drive/MyDrive/Colab Notebooks/csv/iris.csv') #<-change link

 

df['species'].replace({'setosa':a0, 'versicolor':1, 'virginica':2}, inplace=True)

 

#종속변수,독립변수

 

feature_columns = list(df.columns.difference(['species']))

X = df[(feature_columns)]

y = df['species']

 

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=11)

 

#데이터셋 크기확인

print(X_train.shape, X_test.shape, y_train.shape, y_test.shape)

 

#model DecisionTreeClassifier 객체생성

dt = DecisionTreeClassifier(random_state=11)

dt.fit(X_train, y_train) #train

#분류예측수행

pred=dt.predict(X_test)

 

 

#모델성능-정확도측정

from sklearn.metrics import accuracy_score

acc = accuracy_score(y_test, pred)

print(acc)