MACHINE | DEEP LEARNING/Machine Learning Project
[AI] 머신 러닝 Linear Regression 후 메모리 파일 저장 _joblib, 피클파일 저장
처카푸
2024. 4. 14. 21:56
Linear Regression 을 하고 나면,
컴퓨터를 끄고 켰을 때나 다른 서버에서 사용 해야하는 메모리가 있다
다른 서버에서도 내가 만든 Regression이 동작할 수 있도록 해야 한다
메모리에 있는 인공지능을 파일로 저장하여 사용하면 된다
서비스 배포하기 위한 파일 저장 방법
- 임폴트 joblib
import joblib
- 실서버에 인공지능을 활용하려면 2개의 파일이 필요하다
- 데이터를 이용해 내가 학습시킨 인공지능인 regressor
- 문자열로 작성되어 있던 컬럼을 숫자로 바꿔주는 ct _기초 데이터로 교육 시켜져 있다
- joblib를 이용하여 피클 파일로 저장한다
joblib.dump( regressor, 'regressor.pkl')
joblib.dump( ct, 'ct.pkl')
- 서버로 파일을 보내면 예측서비스를 할 수 있다