93 lines
3.6 KiB
Python
93 lines
3.6 KiB
Python
import pandas
|
||
from flask import Flask
|
||
from sklearn.metrics import accuracy_score
|
||
from sklearn.model_selection import train_test_split
|
||
from sklearn.tree import DecisionTreeClassifier
|
||
|
||
app = Flask(__name__)
|
||
|
||
|
||
@app.route("/")
|
||
def home():
|
||
return "<html>" \
|
||
"<h1>Жукова Алина ПИбд-41</h1>" \
|
||
"<h1>Лабораторная работа №3</h1>" \
|
||
"<table>" \
|
||
"<td>" \
|
||
"<form Action='http://127.0.0.1:5000/k4_1_task_3' Method=get>" \
|
||
"<input type=submit value='Дерево решений'>" \
|
||
"</form>" \
|
||
"</td>" \
|
||
"</table>" \
|
||
"</html>"
|
||
|
||
# Деревья решений
|
||
# 3.2 Решите задачу классификации
|
||
# (с помощью дерева решений), в которой по различным характеристикам
|
||
# пассажиров требуется найти у выживших пассажиров два наиболее важных
|
||
# признака из трех рассматриваемых (Name, Cabin,Embarked)
|
||
# 3.2
|
||
#
|
||
#
|
||
@app.route("/k4_1_task_3", methods=['GET'])
|
||
def k4_1_task_3():
|
||
data = pandas.read_csv('Data_chess_games.csv', index_col='id')
|
||
data = data.sample(n=5000, replace=True, random_state=1)
|
||
|
||
# отбор нужных столбцов
|
||
corr = data[['rated', 'turns', 'increment_code', 'white_rating', 'black_rating', 'opening_ply',
|
||
'created_at', 'last_move_at', 'white_id', 'black_id']]
|
||
|
||
# Добавление времени игры
|
||
corr['time_game'] = corr['last_move_at'] - corr['created_at']
|
||
|
||
def new_code(code):
|
||
return (int(code.split("+")[0]) * 100) + int(code.split("+")[1])
|
||
corr['n_increment_code'] = corr['increment_code'].apply(new_code)
|
||
|
||
def strToint(elem):
|
||
it_i = 0
|
||
for ch in elem:
|
||
it_i += ord(ch)
|
||
return 2500 - it_i
|
||
corr['n_white_id'] = corr['white_id'].apply(strToint)
|
||
corr['n_black_id'] = corr['black_id'].apply(strToint)
|
||
|
||
|
||
corr = corr[['white_rating', 'black_rating', 'n_white_id', 'n_black_id']]
|
||
|
||
# определение целевой переменной
|
||
y = data['victory_status']
|
||
|
||
X_train, X_test, y_train, y_test = train_test_split(corr, y, test_size=.01, random_state=42)
|
||
|
||
# создание и обучение дерева решений
|
||
clf = DecisionTreeClassifier(random_state=241)
|
||
clf.fit(X_train, y_train)
|
||
prediction = clf.predict(X_test)
|
||
accuracy = str(accuracy_score(y_test, prediction))
|
||
# получение и распечатка важностей признаков
|
||
importances = clf.feature_importances_
|
||
|
||
data = {}
|
||
ind = 0
|
||
st = []
|
||
for el in importances:
|
||
st.append(el)
|
||
ind += 1
|
||
|
||
data["Важность"] = st
|
||
df = pandas.DataFrame(data, index=corr.columns)
|
||
|
||
return "<html>" \
|
||
"<h1>Деревья решений</h1>" \
|
||
"<h2>Вариант 10. Задание 3.2</h2>" \
|
||
"<h2>Данные об исходе шахматных партий. Необходимо определить исход партии (поставлен мат, кончилось время, соперник сдался, ничья)</h2>" \
|
||
"<h2>Выбраны признаки: Рейтинг игрока за белых, рейтинг игрока за черных, id игроков</h2>" \
|
||
"<h2>Важность признаков: </h2>" \
|
||
"<div>" + df.to_html() + "</div>" \
|
||
"<h2>Точность предсказания: " + str(accuracy) + "</h2>" \
|
||
"</html>"
|
||
|
||
if __name__ == "__main__":
|
||
app.run(debug=True) |