개발학습일지

[AI] 머신 러닝 Linear Regression 후 메모리 파일 저장 _joblib, 피클파일 저장 본문

AI/Machine Learning

[AI] 머신 러닝 Linear Regression 후 메모리 파일 저장 _joblib, 피클파일 저장

처카푸 2024. 4. 14. 21:56

Linear Regression 을 하고 나면,

컴퓨터를 끄고 켰을 때나 다른 서버에서 사용 해야하는 메모리가 있다

다른 서버에서도 내가 만든 Regression이 동작할 수 있도록 해야 한다

메모리에 있는 인공지능을 파일로 저장하여 사용하면 된다

 

서비스 배포하기 위한 파일 저장 방법

- 임폴트 joblib

import joblib

- 실서버에 인공지능을 활용하려면 2개의 파일이 필요하다

   - 데이터를 이용해 내가 학습시킨 인공지능인 regressor

내가 학습시킨 인공지능

   - 문자열로 작성되어 있던 컬럼을 숫자로 바꿔주는 ct  _기초 데이터로 교육 시켜져 있다

내가 학습시킨 ColumnTransformer

 - joblib를 이용하여 피클 파일로 저장한다

joblib.dump( regressor, 'regressor.pkl')
joblib.dump( ct, 'ct.pkl')

 - 서버로 파일을 보내면 예측서비스를 할 수 있다