-
RandomForest 예제 2 - django를 활용(patient 데이터 사용)Python 데이터 분석 2022. 11. 22. 18:12
url.py
from django.contrib import admin from django.urls import path from myapp import views urlpatterns = [ path("admin/", admin.site.urls), path("", views.MainFunc), path("show", views.ShowFunc), path("list", views.ListFunc) ]views.py
from django.shortcuts import render import pandas as pd import numpy as np from sklearn.ensemble import RandomForestClassifier from sklearn.model_selection import train_test_split from sklearn.metrics import accuracy_score import pickle def MainFunc(request): df = pd.read_csv('https://raw.githubusercontent.com/pykwon/python/master/testdata_utf8/patient.csv', index_col=0) df1 = df.head(3) df_x = df.drop(columns = ['STA']) df_y = df['STA'] x_train, x_test, y_train, y_test = train_test_split(df_x, df_y, test_size = 0.2, random_state = 1) print(x_train.shape, x_test.shape, y_train.shape, y_test.shape) # (160, 9) (40, 9) (160, 1) (40, 1) model = RandomForestClassifier(n_estimators=500, criterion='entropy') model.fit(x_train, y_train) pred = model.predict(x_test) acc = accuracy_score(y_test, pred) # 0.85 pickle.dump(model, open('model.sav', mode = 'wb')) del model return render(request, 'main.html', {'df':df1.to_html, 'acc':acc}) def ShowFunc(request): return render(request, 'show.html') def ListFunc(request): import pickle model = pickle.load(open('model.sav', 'rb')) AGE = request.POST.get('AGE') SEX = request.POST.get('SEX') RACE = request.POST.get('RACE') SER = request.POST.get('SER') CAN = request.POST.get('CAN') CRN = request.POST.get('CRN') INF = request.POST.get('INF') CPR = request.POST.get('CPR') HRA = request.POST.get('HRA') data = { AGE:[AGE], SEX:[SEX], RACE:[RACE], SER:[SER], CAN:[CAN], CRN:[CRN], INF:[INF], CPR:[CPR], HRA:[HRA] } x_test = pd.DataFrame(data, columns=[AGE, SEX, RACE, SER, CAN, CRN, INF, CPR, HRA]) print(x_test) print(model) pred = model.predict(x_test) print(pred) return render(request, 'list.html', {'pred':pred[0]})main.html
<!DOCTYPE html> <html> <head> <meta charset="UTF-8"> <title>Insert title here</title> </head> <body> <p>분류모델의 정확도 : {{acc}}</p> <p>patient 데이터 일부 보기 : {{df | safe}}</p> <a href="/show">입원 환자 데이터를 입력해 생사 여부 확인하기</a> </body> </html>
show.html
<!DOCTYPE html> <html> <head> <meta charset="UTF-8"> <title>Insert title here</title> <style> table, th, td { border: 1px solid #bcbcbc; word-break : break-all; } </style> </head> <body> <p>입원 환자 데이터를 입력해 생사 여부 확인을 위한 자료 입력</p> <pre> AGE : 나이 SEX : 성별 RACE : 인종 SER : 중환자 치료실에서 받은 치료 CAN : 암 존재 여부 INF : 중환자 치료실에서의 감염 여부 CPR : 중환자 치료실 도착 전 CPR 여부 HPA : 중환자 치료실에서의 심박수 예) AGE SEX RACE SER CAN INF CPR HPA 27 1 1 0 0 1 0 88 </pre> <form action="/list" method="post" id='list'> {% csrf_token %} <table border="1"> <tr> <th>AGE</th><th>SEX</th><th>RACE</th><th>SER</th> </tr> <tr> <td> <input type="text" id="AGE" name="AGE"/> </td> <td> <input type="text" id="SEX" name="SEX"/> </td> <td> <input type="text" id="RACE" name="RACE"/> </td> <td> <input type="text" id="SER" name="SER"/> </td> </tr> <tr> <th>CAN</th><th>CRN</th><th>INF</th><th>CPR</th><th>HRA</th> </tr> <tr> <td> <input type="text" id="CAN" name="CAN"/> </td> <td> <input type="text" id="CRN" name="CRN"/> </td> <td> <input type="text" id="INF" name="INF"/> </td> <td> <input type="text" id="CPR" name="CPR"/> </td> <td> <input type="text" id="HRA" name="HRA"/> </td> </tr> <tr> <td> <button type="submit">확인</button> </td> </tr> </table> </form> </body> </html>
list.html
<!DOCTYPE html> <html> <head> <meta charset="UTF-8"> <title>Insert title here</title> </head> <body> <p>예측 결과 : {{pred}}</p> {% if pred == 1 %} <p>생존</p> {% else %} <p>사망</p> {% endif %} </body> </html>
'Python 데이터 분석' 카테고리의 다른 글
Python 데이터분석 기초 64 - XGBoost로 분류 모델 작성, lightgbm로 분류 모델 작성 (0) 2022.11.23 Python 데이터분석 기초 63 - DecisionTreeRegressor, RandomForestRegressor (0) 2022.11.23 RandomForest 예제 1 - Red Wine quality 데이터 (0) 2022.11.22 Python 데이터분석 기초 62 - Random forest 예제(titanic) (0) 2022.11.22 Python 데이터분석 기초 61 - Random forest (0) 2022.11.22