728x90
위 사진과 같이 인공지능을 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 라이브러리를 이용해 불러온 후 이용하면 된다.
'streamlit' 카테고리의 다른 글
비주얼스튜디오,주피터노트북 // App을 개발하는 순서. (0) | 2022.12.13 |
---|---|
Visual Studio Code : streamlit을 이용한 차트 그리기 (2) (0) | 2022.12.13 |
Visual Studio Code : streamlit을 이용한 차트 그리기 (1) (0) | 2022.12.13 |
Visual Studio Code : 다른 파일에서 함수 가져와 사용하기 (0) | 2022.12.13 |
Visual Studio Code : 웹 대시보드에 파일을 업로드하는 방법 (0) | 2022.12.13 |