MS AI School
DAY 30 - 모델 저장 및 불러오기 .PKL
🐱👤지식닌자
2022. 12. 6. 21:04
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