Web Dashboard

Jupyter notebook 에서 학습한 인공지능을, streamlit 에서 사용하는 방법

yugyeong 2022. 12. 14. 14:24

 

주피터 노트북에서 학습한 인공지능을 streamlit 에서 사용하는 방법이다

 

 

 

주피터노트북으로 만든 인공지능 regressor 을 joblib 라이브러리를 이용하여서 저장을 해주었다.

인공지능은 'pkl' 형식으로 저장하였다.

 

 

 

 

이제 저장한 인공지능을 streamlit 으로 불러오는 방법이다.

 

streamlit 은 비주얼스튜디오 코드에서 실행을 시키는 것이기 때문에, vs code 에서 joblib 라이브러리를 임포트 해준다.

joblib.load 를 이용하면 저장한 인공지능을 불러올 수 있다.

 

import joblib

regressor = joblib.load('regressor.pkl')

 

저장한 파일을 load 하면 인공지능이 streamlit 에 불러와진 것이다.