728x90
# 번외편 PKL 파일 저장 및 로딩
from sklearn import datasets
from sklearn.neighbors import KNeighborsClassifier
import joblib
dataset = datasets.load_iris()
X, y = dataset['data'], dataset['target']
model = KNeighborsClassifier(n_neighbors=3)
model.fit(X, y)
joblib.dump(model, './knn_model.pkl')
loaded_model = joblib.load('./knn_model.pkl')
loaded_model.predict
score = loaded_model.score(X, y)
print('정확도: {score:.3f}'.format(score=score))
728x90
'MS AI School' 카테고리의 다른 글
DAY 32 - 다중 분류 실습(로이터 기사) (0) | 2022.12.06 |
---|---|
DAY 31 - 이진 분류 실습(영화 리뷰 IMDB dataset) (0) | 2022.12.06 |
DAY 29 - Docker 도커? (0) | 2022.12.06 |
DAY 28 - 딥러닝 Training 과정 (0) | 2022.12.06 |
DAY 27 - 딥러닝 프레임워크 Keras (0) | 2022.11.29 |