python 간단한 ai 코드
2023. 7. 1. 22:00ㆍProgramming/python
python 간단한 ai 코드 위해서
sklearn 설치해야 합니다.
SLEARN 설치
pip install scikit-learn
MAIN CODE
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.neighbors import KNeighborsClassifier
from sklearn.metrics import accuracy_score
# 데이터 로딩
iris = load_iris()
# 데이터 분할 및 전처리
X_train, X_test, y_train, y_test = train_test_split(iris.data, iris.target, test_size=0.2, stratify=iris.target, random_state=42)
# KNN 분류 모델 생성 및 학습
knn = KNeighborsClassifier(n_neighbors=3)
knn.fit(X_train, y_train)
# 예측과 정확도 평가
y_pred = knn.predict(X_test)
accuracy = accuracy_score(y_test, y_pred)
# 결과 출력
print("Accuracy: {:.2f}%".format(accuracy * 100))
위 코드에서는 iris 데이터셋을 사용하여 꽃의 종류를 분류하는 문제를 해결하는 코드입니다. scikit-learn에서 제공하는 load_iris() 함수를 사용하여 데이터를 불러오고, train_test_split() 함수를 사용하여 데이터를 훈련용과 테스트용으로 분할합니다. 분류 문제를 해결하기 위해 KNN(K-Nearest Neighbor) 분류기를 사용하였으며, KNeighborsClassifier() 생성자 함수를 사용하여 모델을 생성하고 fit() 메소드를 사용하여 훈련 데이터로 모델을 학습시킵니다. 학습이 끝난 모델을 predict() 메소드를 이용하여 테스트 데이터에 대해서 예측합니다. 예측한 결과를 accuracy_score() 함수를 사용하여 정확도를 측정합니다. 최종적으로 결과를 출력합니다. 위 코드를 실행시켜 보면, 예측한 결과의 정확도가 출력됩니다. 해당 코드에서는 iris 데이터셋을 사용하였지만, 다른 데이터셋을 사용하여도 비슷한 방식으로 데이터를 다룰 수 있습니다. 이 코드를 기반으로 scikit-learn을 사용하여 더 다양한 분류 문제를 해결하는 코드를 작성할 수 있을 것입니다.
놀랍고 굉장한 출력이 있는건 아니다.
Accuracy: 100.00%
흠....
반응형
'Programming > python' 카테고리의 다른 글
파이썬 랜덤 번호 추출하기(Python Tutorial: Building a Random Number Generator) (3) | 2024.11.12 |
---|---|
Pytrhon 설치없이 사용하기 - Colaboratory(코랩) (0) | 2023.07.02 |
파이썬 로그 남기기 Logging(Log, Logger, Logging) (0) | 2023.07.01 |
파이썬 환경변수 설정- autocomplete-python traceback output: (0) | 2023.07.01 |
The Python Tutorial - Setting up the Atom for python (0) | 2023.07.01 |