create README
This commit is contained in:
parent
676080d48d
commit
5a83f61bd4
BIN
istyukov_timofey_lab_3/1_dataset.jpg
Normal file
BIN
istyukov_timofey_lab_3/1_dataset.jpg
Normal file
Binary file not shown.
After Width: | Height: | Size: 45 KiB |
BIN
istyukov_timofey_lab_3/2_accuracy_score.jpg
Normal file
BIN
istyukov_timofey_lab_3/2_accuracy_score.jpg
Normal file
Binary file not shown.
After Width: | Height: | Size: 41 KiB |
BIN
istyukov_timofey_lab_3/3_feature_importances.jpg
Normal file
BIN
istyukov_timofey_lab_3/3_feature_importances.jpg
Normal file
Binary file not shown.
After Width: | Height: | Size: 16 KiB |
73
istyukov_timofey_lab_3/README.md
Normal file
73
istyukov_timofey_lab_3/README.md
Normal file
@ -0,0 +1,73 @@
|
||||
# Лабораторная работа №3. Деревья решений
|
||||
## 12 вариант
|
||||
___
|
||||
|
||||
### Задание:
|
||||
Решите с помощью библиотечной реализации дерева решений задачу из
|
||||
лабораторной работы «Веб-сервис «Дерево решений» по предмету
|
||||
«Методы искусственного интеллекта» на 99% ваших данных.
|
||||
Проверьте работу модели на оставшемся проценте, сделайте вывод.
|
||||
|
||||
|
||||
### Вариант набора данных по курсовой работе:
|
||||
- Прогнозирование музыкальных жанров
|
||||
|
||||
___
|
||||
|
||||
### Запуск
|
||||
- Запустить файл lab3.py
|
||||
|
||||
### Используемые технологии
|
||||
- Язык программирования **Python**
|
||||
- Среда разработки **PyCharm**
|
||||
- Библиотеки:
|
||||
* pandas
|
||||
* sklearn
|
||||
|
||||
### Описание программы
|
||||
**Набор данных (Kaggle):** Полный список жанров, включенных в CSV:
|
||||
«Электронная музыка», «Аниме», «Джаз», «Альтернатива», «Кантри», «Рэп»,
|
||||
«Блюз», «Рок», «Классика», «Хип-хоп».
|
||||
|
||||
**Задача, решаемая деревом решений:** Классификация музыкальных треков на
|
||||
основе их характеристик, таких как темп, инструментальность, акустичность,
|
||||
речевость, танцевальность, энергичность, живость. Дерево решений может
|
||||
предсказывать жанр трека, основываясь на его характеристиках.
|
||||
|
||||
**Задачи оценки:** оценить качество работы модели дерева решений и выявить
|
||||
наиболее значимые признаки набора данных.
|
||||
|
||||
---
|
||||
### Пример работы
|
||||
|
||||
*Датасет, сформированный из случайных строк csv-файла.*
|
||||
![Graphics](1_dataset.jpg)
|
||||
|
||||
---
|
||||
*Сравнение на оставшихся неиспользованных 0,5% строк датасета
|
||||
предсказнных и действительных жанров.*
|
||||
|
||||
![Graphics](2_accuracy_score.jpg)
|
||||
|
||||
---
|
||||
*Вычисленнные коэффициенты влияния признаков на прогноз жанра*
|
||||
|
||||
![Graphics](3_feature_importances.jpg)
|
||||
|
||||
---
|
||||
|
||||
### Вывод
|
||||
Посредством предобработки датасета дерево решений без проблем обучилось и
|
||||
частично верно предсказало некоторые жанры (в частности, Электро, Классику
|
||||
и Рэп). Также модель показала оценку влиятельности признаков на прогноз
|
||||
жанра. Самым влиятельным признаком оказалась **акустичность** музыкального
|
||||
трека. Менее значимыми оказались речевость (преобладание голосов в треке) и
|
||||
инструментальность (преобладание живых инструментов в треке), что звучит
|
||||
вполне разумно.
|
||||
|
||||
На практике дерево решений по качеству классификации уступает некоторым
|
||||
другим методам. Помимо этого, небольшие изменения в данных могут существенно
|
||||
изменять построенное дерево решений. На примере моего датасета дерево решений
|
||||
справилось не очень успешно. Это можно объяснить тем, что данных в нём
|
||||
недостаточно для предсказания жанра. Но также стоит отметить, что
|
||||
жанр – одно из самых неоднозначных, самых многосоставных музыкальных понятий.
|
69
istyukov_timofey_lab_3/lab3.py
Normal file
69
istyukov_timofey_lab_3/lab3.py
Normal file
@ -0,0 +1,69 @@
|
||||
"""
|
||||
Решите с помощью библиотечной реализации дерева решений задачу изnлабораторной работы
|
||||
«Веб-сервис «Дерево решений» по предмету «Методы искусственного интеллекта» на 99% ваших данных.
|
||||
Проверьте работу модели на оставшемся проценте, сделайте вывод.
|
||||
"""
|
||||
|
||||
"""
|
||||
Задача, решаемая деревом решений: Классификация музыкальных треков на основе их характеристик,
|
||||
таких как акустика, танцевальность, инструментальность, темп и т.д.
|
||||
Дерево решений может предсказывать жанр трека, основываясь на его характеристиках.
|
||||
"""
|
||||
|
||||
# 12 вариант
|
||||
# Набор данных по курсовой: "Prediction of music genre"
|
||||
|
||||
import pandas as pd
|
||||
from sklearn.tree import DecisionTreeClassifier
|
||||
from sklearn.model_selection import train_test_split
|
||||
from sklearn.metrics import accuracy_score
|
||||
|
||||
|
||||
DATASET_FILE = 'music_genre.csv'
|
||||
|
||||
|
||||
def main():
|
||||
df = open_dataset(DATASET_FILE)
|
||||
df = df.sample(frac=.1) # отбираем 10% рандомных строк с набора данных, т.к. он большой
|
||||
print("\033[92m[-----> Набор данных <-----]\033[00m")
|
||||
print(df)
|
||||
|
||||
X = df.drop(columns=['music_genre']) # набор числовых признаков
|
||||
y = df['music_genre'] # набор соответствующих им жанров
|
||||
|
||||
# Разделение датасета на тренировочные (99,5%) и тестовые данные (0,5%)
|
||||
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.005)
|
||||
|
||||
# Создание и обучение дерева решений
|
||||
model = DecisionTreeClassifier()
|
||||
model.fit(X_train.values, y_train)
|
||||
|
||||
# Прогнозирование жанра на тестовых данных
|
||||
y_pred = model.predict(X_test.values)
|
||||
|
||||
print("\033[92m\n\n\n[-----> Сравнение жанров <-----]\033[00m")
|
||||
df_result = pd.DataFrame({'Прогноз': y_pred, 'Реальность': y_test})
|
||||
print(df_result)
|
||||
|
||||
score = accuracy_score(y_test, y_pred)
|
||||
print("\033[92m\n> Оценка точности модели: {}\033[00m" .format(round(score, 2)))
|
||||
|
||||
print("\033[92m\n\n\n[-----> Оценки важности признаков <-----]\033[00m")
|
||||
df_feature = pd.DataFrame({'Признак': X.columns, "Важность": model.feature_importances_})
|
||||
print(df_feature)
|
||||
|
||||
|
||||
# Функция считывания и очищения csv-файла
|
||||
def open_dataset(csv_file):
|
||||
# открываем файл с указанием знака-отделителя
|
||||
df_genres = pd.read_csv(csv_file, delimiter=',')
|
||||
# выбираем необходимые признаки
|
||||
df_genres = df_genres[['tempo', 'instrumentalness', 'acousticness', 'speechiness', 'danceability', 'energy', 'liveness', 'music_genre']]
|
||||
# очищаем набор данных от пустых и неподходящих значений
|
||||
df_genres = df_genres[df_genres['tempo'] != '?']
|
||||
df_genres = df_genres.dropna()
|
||||
return df_genres
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
50006
istyukov_timofey_lab_3/music_genre.csv
Normal file
50006
istyukov_timofey_lab_3/music_genre.csv
Normal file
File diff suppressed because it is too large
Load Diff
Loading…
Reference in New Issue
Block a user