kurmyza_pavel_lab_3 is ready
This commit is contained in:
parent
a98d914e7c
commit
6a9602359c
55
kurmyza_pavel_lab_3/README.md
Normal file
55
kurmyza_pavel_lab_3/README.md
Normal file
@ -0,0 +1,55 @@
|
||||
# Лабораторная работа №3
|
||||
|
||||
## ПИбд-41, Курмыза Павел
|
||||
|
||||
Датасет по варианту: https://www.kaggle.com/datasets/jessemostipak/hotel-booking-demand.
|
||||
|
||||
Данный набор данных содержит информацию о бронировании городской и курортной гостиниц и включает в себя такие
|
||||
сведения, как время бронирования, продолжительность пребывания, количество взрослых, детей и/или младенцев, количество
|
||||
свободных парковочных мест и т.д.
|
||||
|
||||
## Как запустить ЛР
|
||||
|
||||
- Запустить файл main.py
|
||||
|
||||
## Используемые технологии
|
||||
|
||||
- Язык программирования Python
|
||||
- Библиотеки: sklearn, numpy, pandas, xgboost, matplotlib, seaborn
|
||||
|
||||
## Что делает программа
|
||||
|
||||
Программа решает задачу классификации на выбранном датасете: определение гостиничного класса отеля (городской отель или
|
||||
курортный отель). Решение достигается в несколько этапов:
|
||||
|
||||
- Предобработка данных
|
||||
- Балансировка данных
|
||||
- Стандартизация данных и приведение их к виду, удобном для работы с моделями ML
|
||||
- Использование нескольких моделей классификации
|
||||
- Сравнение оценок и поиск наиболее подходящей модели
|
||||
- Оценка специфичности наилучшей модели классификации
|
||||
|
||||
## Тестирование
|
||||
|
||||
Для решения задачи классификации были выбраны 3 модели: XGBClassifier, RandomForestClassifier, DecisionTreeClassifier.
|
||||
|
||||
Оценка точности моделей:
|
||||
|
||||
- XGBClassifier: 0.970565942395149
|
||||
- RandomForestClassifier: 0.9901465386558869
|
||||
- DecisionTreeClassifier: 0.9714502273875695
|
||||
|
||||
Оцека способности модели RandomForestClassifier предсказывать истинные положительные результаты (TP / (TP + FN)), также
|
||||
известные как коэффициент чувствительности, и истинные отрицательные результаты (TN / (TN + FP)), также известный как
|
||||
коэффициент специфичности через матрицу неточностей:
|
||||
|
||||
![Матрица неточностей](confusion_matrix.jpg)
|
||||
|
||||
Матрица неточностей подтверждает приведенную ранее оценку модели RandomForestClassifier. Кроме того, она указывает на
|
||||
то, что помимо высокой точности, модель также имеет высокую специфичность.
|
||||
|
||||
## Вывод
|
||||
|
||||
По итогу тестирования было выявлено, что наилучше всего с задачей классификации данного набора данных справляется модель
|
||||
RandomForestClassifier, так как имеет наивысшую оценку точности. Тем не менее, все из выбранных моделей показали высокие
|
||||
результаты. Из этого следует, что задача классификации для данного набора решена.
|
BIN
kurmyza_pavel_lab_3/confusion_matrix.jpg
Normal file
BIN
kurmyza_pavel_lab_3/confusion_matrix.jpg
Normal file
Binary file not shown.
After Width: | Height: | Size: 26 KiB |
119391
kurmyza_pavel_lab_3/hotel_bookings.csv
Normal file
119391
kurmyza_pavel_lab_3/hotel_bookings.csv
Normal file
File diff suppressed because it is too large
Load Diff
119
kurmyza_pavel_lab_3/main.py
Normal file
119
kurmyza_pavel_lab_3/main.py
Normal file
@ -0,0 +1,119 @@
|
||||
import pandas as pd
|
||||
from matplotlib import pyplot as plt
|
||||
from sklearn.tree import DecisionTreeClassifier
|
||||
from sklearn.ensemble import RandomForestClassifier
|
||||
from sklearn.preprocessing import LabelEncoder
|
||||
from sklearn.feature_selection import VarianceThreshold
|
||||
from sklearn.model_selection import train_test_split
|
||||
from sklearn.preprocessing import StandardScaler
|
||||
from xgboost import XGBClassifier
|
||||
from sklearn.metrics import confusion_matrix, classification_report
|
||||
import seaborn as sns
|
||||
|
||||
# Считываем датасет
|
||||
ds = pd.read_csv('hotel_bookings.csv')
|
||||
|
||||
# Удаляем из датасета строки с пропущенными значениями столбцов country, children.
|
||||
# Выбраны именно данные столбцы, так как, по информации из kaggle, только они могут содержать пропущеные значения
|
||||
ds.dropna(axis=0, subset=['country', 'children'], inplace=True)
|
||||
|
||||
# Усредняем значения столбца agent, чтобы убрать его влияние на результат, так как столбец содержит неважные данные
|
||||
moa = ds['agent'].mean()
|
||||
ds['agent'].fillna(value=moa, axis=0, inplace=True)
|
||||
|
||||
# Заполняем пропущенные значения ячеек, чтобы исключить незаполненные
|
||||
ds.fillna(method='pad', inplace=True)
|
||||
ds.dropna(inplace=True, subset=['company'])
|
||||
|
||||
# Переводим столбцы, содержащие текстовые данные в числовое представление
|
||||
hotel = LabelEncoder()
|
||||
meal = LabelEncoder()
|
||||
country = LabelEncoder()
|
||||
market_segment = LabelEncoder()
|
||||
distribution_channel = LabelEncoder()
|
||||
reserved_room_type = LabelEncoder()
|
||||
assigned_room_type = LabelEncoder()
|
||||
deposit_type = LabelEncoder()
|
||||
customer_type = LabelEncoder()
|
||||
reservation_status = LabelEncoder()
|
||||
reservation_status_date = LabelEncoder()
|
||||
|
||||
ds['hotel_n'] = hotel.fit_transform(ds['hotel'])
|
||||
ds['arrival_date_month_n'] = hotel.fit_transform(ds['arrival_date_month'])
|
||||
ds['meal_n'] = hotel.fit_transform(ds['meal'])
|
||||
ds['country_n'] = hotel.fit_transform(ds['country'])
|
||||
ds['market_segment_n'] = hotel.fit_transform(ds['market_segment'])
|
||||
ds['distribution_channel_n'] = hotel.fit_transform(ds['distribution_channel'])
|
||||
ds['reserved_room_type_n'] = hotel.fit_transform(ds['reserved_room_type'])
|
||||
ds['assigned_room_type_n'] = hotel.fit_transform(ds['assigned_room_type'])
|
||||
ds['deposit_type_n'] = hotel.fit_transform(ds['deposit_type'])
|
||||
ds['customer_type_n'] = hotel.fit_transform(ds['customer_type'])
|
||||
ds['reservation_status_n'] = hotel.fit_transform(ds['reservation_status'])
|
||||
ds['reservation_status_date_n'] = hotel.fit_transform(ds['reservation_status_date'])
|
||||
|
||||
# Удаляем приведенные к числовым данным столбцы, они больше не нужны
|
||||
ds.drop(
|
||||
['hotel', 'arrival_date_month', 'meal', 'country', 'market_segment', 'distribution_channel', 'reserved_room_type',
|
||||
'assigned_room_type', 'deposit_type', 'customer_type', 'reservation_status', 'reservation_status_date'], axis=1,
|
||||
inplace=True)
|
||||
|
||||
# Производим балансировку данных таким образом, чтобы было одинаковое количество отелей всех классов
|
||||
ds_0 = ds[ds['hotel_n'] == 0]
|
||||
ds_1 = ds[ds['hotel_n'] == 1]
|
||||
ds_0 = ds_0.sample(ds_1.shape[0])
|
||||
ds = ds_0._append(ds_1, ignore_index=True)
|
||||
|
||||
# Полдготовка данных для выполнения модели
|
||||
x = ds.drop('hotel_n', axis=1)
|
||||
y = ds['hotel_n']
|
||||
|
||||
threshold = VarianceThreshold()
|
||||
|
||||
x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.2)
|
||||
x_train = threshold.fit_transform(x_train)
|
||||
x_test = threshold.transform(x_test)
|
||||
|
||||
# Производим стандартизацию данных и приводим их к виду, с которым работают модели классификации
|
||||
scaler = StandardScaler()
|
||||
|
||||
x_train = scaler.fit_transform(x_train)
|
||||
x_test = scaler.fit_transform(x_test)
|
||||
|
||||
y_train = y_train.to_numpy()
|
||||
y_test = y_test.to_numpy()
|
||||
|
||||
# Использование моделей машинного обучения.
|
||||
# Для сравнения точности были взяты модели XGBClassifier, RandomForestClassifier, DecisionTreeClassifier.
|
||||
|
||||
xg = XGBClassifier()
|
||||
xg.fit(x_train, y_train)
|
||||
|
||||
rf = RandomForestClassifier()
|
||||
rf.fit(x_train, y_train)
|
||||
|
||||
dt = DecisionTreeClassifier()
|
||||
dt.fit(x_train, y_train)
|
||||
|
||||
# Оценка точности моделей классификации
|
||||
|
||||
xgb_accuracy = xg.score(x_test, y_test)
|
||||
rf_accuracy = rf.score(x_test, y_test)
|
||||
dt_accuracy = dt.score(x_test, y_test)
|
||||
|
||||
print(f"Оценка точности моделей:"
|
||||
f"\n XGBClassifier: {xgb_accuracy}",
|
||||
f"\n RandomForestClassifier: {rf_accuracy}",
|
||||
f"\n DecisionTreeClassifier: {dt_accuracy}")
|
||||
|
||||
# Оценка коэффициента специфичности через матрицу неточностей
|
||||
|
||||
y_pred = rf.predict(x_test)
|
||||
|
||||
cm = confusion_matrix(y_test, y_pred)
|
||||
plt.figure(figsize=(7, 5))
|
||||
sns.heatmap(cm, annot=True)
|
||||
plt.xlabel('Prediction')
|
||||
plt.ylabel('Actual')
|
||||
plt.show()
|
||||
|
||||
print(classification_report(y_test, y_pred))
|
Loading…
Reference in New Issue
Block a user