Visual Studio Code : jupyter notebook에서 학습한 인공지능을, 스트림릿에서 사용하는 방법.

위 사진과 같이 인공지능을 jupyter notebook에서 만들어준다.

 

 

이후

 

 

import streamlit as st
import numpy as np
import joblib
regressor = joblib.load('regressor.pkl')

def run_ml_app():
    st.subheader('자동차 금액 예측')


    # 성별, 나이, 연봉, 카드빚, 자산을 유저한테 모두 입력 받아서
    # 자동차 구매 금액 예측하세요.

    gender = st.radio('성별 선택',['여자','남자'])
    if gender == '여자':
        gender = 0
    else :
        gender = 1

    age = st.number_input('나이 입력',18,100)

    salary = st.number_input('연봉 입력',10000,1000000)

    debt = st.number_input('카드빚 입력',0,1000000)

    worth = st.number_input('자산 입력',1000,1000000)


    new_data=np.array([gender,age,salary,debt,worth])

    new_data=new_data.reshape(1,5)

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

    y_pred=regressor.predict(new_data)
    y_pred = round(y_pred[0],1)
    if y_pred < 0 :
        st.warning('입력한 데이터로는 금액을 예측하기 어렵습니다.')
    else :
        st.info('예측한 자동차 금액은 {}달러 입니다.'.format(y_pred))

joblib 라이브러리를 이용해 불러온 후 이용하면 된다.