Compare commits
2 Commits
orlov_arte
...
mashkova_m
| Author | SHA1 | Date | |
|---|---|---|---|
| 704d6631fe | |||
| a8c58683dd |
118
kutygin_andre_lab_3/README.md
Normal file
118
kutygin_andre_lab_3/README.md
Normal file
@@ -0,0 +1,118 @@
|
||||
**Задание**
|
||||
***
|
||||
Решите с помощью библиотечной реализации дерева решений задачу из лабораторной работы «Веб-сервис «Дерево решений» по предмету «Методы искусственного интеллекта»на 99% ваших данных. Проверьте работу модели на оставшемся проценте, сделайте вывод
|
||||
|
||||
**Как запустить лабораторную**
|
||||
***
|
||||
Запустить файл main.py
|
||||
|
||||
**Используемые технологии**
|
||||
***
|
||||
Библиотеки pandas, scikit-learn, matplotlib, их компоненты
|
||||
|
||||
**Описание лабораторной (программы)**
|
||||
***
|
||||
В данном коде мы создаем и обучаем модель дерева решений для прогнозирования инцидентов с НЛО на основе набора данных.
|
||||
|
||||
1. В первой строке кода мы загружаем данные из CSV-файла 'ufo_data_nuforc.csv' с помощью функции pd.read_csv(). Эти данные содержат информацию о различных инцидентах с НЛО.
|
||||
2. Далее мы выбираем набор признаков, в данном случае, эти признаки - населенность и время, которые будут использоваться для обучения модели, и сохраняем их в переменную features.
|
||||
3. Затем преобразуем категориальные признаки в числовой вид при помощи функции pd.get_dummies(). Это необходимо, так как модель дерева решений работает только с числовыми данными.
|
||||
4. После этого мы разделяем данные на обучающую и тестовую выборки с помощью функции train_test_split(). Обучающая выборка будет использоваться для обучения модели, а тестовая - для проверки ее точности.
|
||||
5. Создаем модель дерева решений с помощью класса DecisionTreeClassifier() из библиотеки sklearn.tree.
|
||||
6. Обучаем модель на обучающей выборке с помощью метода fit(). В процессе обучения модель настраивает параметры дерева решений, чтобы лучше предсказывать целевой признак.
|
||||
7. После обучения модели, мы производим прогнозы на тестовых данных с помощью метода predict().
|
||||
8. Оцениваем точность модели на тестовой выборке с помощью метода accuracy_score() из библиотеки sklearn.metrics. Этот метод сравнивает фактические значения целевого признака с предсказанными и возвращает точность модели.
|
||||
9. Наконец, выводим точность модели на тестовой выборке, чтобы оценить, насколько хорошо модель предсказывает инциденты с НЛО.
|
||||
10. Также, код визуализирует данные в виде графика с помощью библиотеки matplotlib.pyplot, отображая фактические значения целевого признака и предсказания модели. Это помогает наглядно оценить, насколько близки предсказания модели к реальным значениям.
|
||||
**Результат**
|
||||
***
|
||||
Точность модели на тестовой выборке: 0.1377245508982036
|
||||
Прогнозы по оставшемуся проценту данных: 'cylinder' 'circle' 'sphere' 'disk' 'disk' 'fireball' 'disk' 'oval'
|
||||
'circle' 'disk' 'disk' 'other' 'light' 'light' 'oval' 'fireball' 'light'
|
||||
'rectangle' 'chevron' 'unknown' 'sphere' 'oval' 'light' 'circle'
|
||||
'unknown' 'unknown' 'disk' 'triangle' 'triangle' 'unknown' 'formation'
|
||||
'unknown' 'cigar' 'unknown' 'light' 'other' 'rectangle' 'light' 'other'
|
||||
'light' 'cylinder' 'delta' 'sphere' 'other' 'changing' 'fireball'
|
||||
'cylinder' 'cigar' 'circle' 'triangle' 'light' 'fireball' 'fireball'
|
||||
'sphere' 'circle' 'light' 'chevron' 'oval' 'oval' 'light' 'unknown'
|
||||
'triangle' 'other' 'rectangle' 'triangle' 'triangle' 'flash' 'unknown'
|
||||
'sphere' 'unknown' 'other' 'circle' 'oval' 'light' 'oval' 'formation'
|
||||
'sphere' 'triangle' 'changing' 'sphere' 'oval' 'unknown' 'circle'
|
||||
'circle' 'flash' 'light' 'light' 'sphere' 'other' 'other' 'egg' 'unknown'
|
||||
'other' 'light' 'light' 'disk' 'diamond' 'oval' 'unknown' 'light'
|
||||
'triangle' 'other' 'light' 'disk' 'unknown' 'light' 'changing' 'sphere'
|
||||
'triangle' 'circle' 'flash' 'sphere' 'light' 'unknown' 'oval' 'formation'
|
||||
'light' 'circle' 'unknown' 'other' 'triangle' 'other' 'light' 'disk'
|
||||
'formation' 'oval' 'triangle' 'triangle' 'light' 'formation' 'oval'
|
||||
'light' 'light' 'oval' 'disk' 'sphere' 'egg' 'unknown' 'unknown'
|
||||
'unknown' 'light' 'disk' 'changing' 'light' 'light' 'circle' 'circle'
|
||||
'formation' 'light' 'light' 'cigar' 'light' 'triangle' 'oval' 'fireball'
|
||||
'cylinder' 'other' 'circle' 'egg' 'changing' 'triangle' 'circle' 'other'
|
||||
'oval' 'disk' 'light' 'flash' 'fireball' 'circle' 'circle' 'circle'
|
||||
'circle' 'light' 'disk' 'fireball' 'other' 'sphere' 'light' 'changing'
|
||||
'cigar' 'light' 'cylinder' 'rectangle' 'chevron' 'light' 'light' 'light'
|
||||
'light' 'circle' 'circle' 'light' 'light' 'circle' 'sphere' 'triangle'
|
||||
'light' 'egg' 'circle' 'fireball' 'sphere' 'sphere' 'triangle' 'light'
|
||||
'other' 'cigar' 'sphere' 'sphere' 'fireball' 'light' 'light' 'disk'
|
||||
'oval' 'oval' 'other' 'cigar' 'triangle' 'light' 'light' 'light' 'disk'
|
||||
'light' 'light' 'light' 'light' 'other' 'light' 'teardrop' 'triangle'
|
||||
'teardrop' 'fireball' 'sphere' 'cylinder' 'fireball' 'circle' 'egg'
|
||||
'sphere' 'disk' 'chevron' 'triangle' 'light' 'other' 'light' 'circle'
|
||||
'rectangle' 'fireball' 'formation' 'light' 'light' 'circle' 'light'
|
||||
'light' 'formation' 'light' 'triangle' 'light' 'oval' 'light' 'unknown'
|
||||
'fireball' 'diamond' 'light' 'circle' 'light' 'triangle' 'oval' 'oval'
|
||||
'cylinder' 'circle' 'light' 'disk' 'light' 'sphere' 'circle' 'light'
|
||||
'triangle' 'light' 'fireball' 'triangle' 'light' 'flash' 'triangle' 'egg'
|
||||
'disk' 'oval' 'circle' 'flash' 'light' 'oval' 'sphere' 'light' 'triangle'
|
||||
'other' 'chevron' 'other' 'circle' 'unknown' 'unknown' 'sphere' 'light'
|
||||
'cigar' 'light' 'fireball' 'circle' 'diamond' 'fireball' 'triangle'
|
||||
'diamond' 'sphere' 'circle' 'chevron' 'cylinder' 'light' 'circle'
|
||||
'fireball' 'unknown' 'light' 'circle' 'fireball' 'light' 'fireball'
|
||||
'fireball' 'fireball' 'light' 'sphere' 'light' 'sphere' 'sphere'
|
||||
'formation' 'light' 'fireball' 'fireball' 'disk' 'disk' 'circle'
|
||||
'rectangle' 'unknown' 'disk' 'unknown' 'disk' 'triangle' 'other' 'sphere'
|
||||
'diamond' 'light' 'light' 'unknown' 'sphere' 'circle' 'disk' 'circle'
|
||||
'oval' 'changing' 'other' 'other' 'disk' 'unknown' 'unknown' 'disk'
|
||||
'rectangle' 'disk' 'light' 'oval' 'unknown' 'sphere' 'light' 'changing'
|
||||
'disk' 'disk' 'other' 'other' 'disk' 'cylinder' 'disk' 'rectangle'
|
||||
'light' 'disk' 'disk' 'light' 'fireball' 'formation' 'cigar' 'oval'
|
||||
'fireball' 'unknown' 'disk' 'light' 'light' 'triangle' 'triangle' 'light'
|
||||
'sphere' 'triangle' 'sphere' 'circle' 'light' 'oval' 'oval' 'circle'
|
||||
'oval' 'rectangle' 'disk' 'oval' 'light' 'light' 'other' 'cigar'
|
||||
'triangle' 'disk' 'cigar' 'other' 'triangle' 'egg' 'unknown' 'triangle'
|
||||
'light' 'triangle' 'disk' 'changing' 'triangle' 'disk' 'disk' 'rectangle'
|
||||
'other' 'triangle' 'triangle' 'formation' 'triangle' 'egg' 'sphere'
|
||||
'fireball' 'triangle' 'rectangle' 'light' 'triangle' 'triangle' 'other'
|
||||
'light' 'light' 'disk' 'fireball' 'light' 'disk' 'oval' 'triangle'
|
||||
'other' 'fireball' 'light' 'light' 'triangle' 'unknown' 'cigar' 'light'
|
||||
'unknown' 'chevron' 'formation' 'disk' 'cigar' 'light' 'sphere' 'cigar'
|
||||
'unknown' 'triangle' 'other' 'light' 'light' 'triangle' 'diamond' 'light'
|
||||
'triangle' 'oval' 'changing' 'light' 'flash' 'circle' 'oval' 'other'
|
||||
'sphere' 'circle' 'triangle' 'unknown' 'teardrop' 'unknown' 'fireball'
|
||||
'light' 'light' 'cigar' 'cigar' 'light' 'fireball' 'other' 'egg' 'light'
|
||||
'other' 'unknown' 'unknown' 'changing' 'circle' 'light' 'other' 'unknown'
|
||||
'unknown' 'light' 'other' 'light' 'unknown' 'cylinder' 'triangle'
|
||||
'circle' 'light' 'circle' 'circle' 'circle' 'light' 'light' 'changing'
|
||||
'changing' 'circle' 'circle' 'triangle' 'triangle' 'light' 'light'
|
||||
'light' 'light' 'other' 'changing' 'triangle' 'cylinder' 'light'
|
||||
'unknown' 'circle' 'disk' 'sphere' 'oval' 'formation' 'teardrop'
|
||||
'triangle' 'chevron' 'light' 'unknown' 'unknown' 'other' 'egg' 'circle'
|
||||
'oval' 'cigar' 'unknown' 'chevron' 'oval' 'cigar' 'fireball' 'circle'
|
||||
'unknown' 'light' 'sphere' 'fireball' 'changing' 'light' 'circle'
|
||||
'unknown' 'fireball' 'light' 'sphere' 'light' 'formation' 'circle'
|
||||
'fireball' 'formation' 'formation' 'formation' 'light' 'other' 'light'
|
||||
'light' 'circle' 'diamond' 'oval' 'circle' 'oval' 'triangle' 'light'
|
||||
'disk' 'light' 'other' 'triangle' 'triangle' 'cylinder' 'disk' 'cylinder'
|
||||
'light' 'oval' 'cigar' 'circle' 'disk' 'light' 'unknown' 'circle' 'other'
|
||||
'light' 'light' 'light' 'unknown' 'triangle' 'other' 'disk' 'cylinder'
|
||||
'triangle' 'oval' 'disk' 'light' 'triangle' 'circle' 'light' 'other'
|
||||
'light' 'other' 'circle' 'disk' 'other' 'triangle' 'oval' 'unknown'
|
||||
'light' 'triangle' 'unknown' 'circle' 'unknown' 'light' 'fireball'
|
||||
'fireball' 'rectangle' 'light' 'formation' 'unknown' 'light' 'light'
|
||||
'formation' 'fireball' 'light' 'light' 'other' 'unknown' 'light'
|
||||
'triangle' 'fireball' 'triangle' 'triangle' 'flash' 'circle' 'triangle'
|
||||
'disk' 'light' 'unknown' 'light' 'light' 'fireball' 'circle' 'unknown'
|
||||
'unknown' 'circle' 'disk' 'chevron' 'disk' 'disk' 'triangle' 'light'
|
||||
'light' 'disk'
|
||||
|
||||
***Вывод:*** Наша модель дерева решений показала низкую точность предсказаний (Точность модели на тестовой выборке: 0.1377245508982036), что означает, что она не очень хорошо предсказывает форму НЛО на основе выбранных признаков (население и время). Из-за чего можно сделать вывод, что возможно, эти признаки недостаточно информативны или недостаточно связаны с формой НЛО.
|
||||
39
kutygin_andre_lab_3/main.py
Normal file
39
kutygin_andre_lab_3/main.py
Normal file
@@ -0,0 +1,39 @@
|
||||
import pandas as pd
|
||||
from sklearn.tree import DecisionTreeClassifier
|
||||
from sklearn.metrics import accuracy_score
|
||||
from sklearn.model_selection import train_test_split
|
||||
|
||||
# Загрузка данных
|
||||
data = pd.read_csv('ufo_sighting_data.csv')
|
||||
|
||||
# Выбор признаков
|
||||
features = [ 'length_of_encounter_seconds', 'latitude', 'longitude']
|
||||
target = 'UFO_shape'
|
||||
# Удаление строк содержащих NaN
|
||||
data.dropna(inplace=True)
|
||||
|
||||
# Удаление столбцов содержащих NaN
|
||||
data.dropna(axis='columns', inplace=True)
|
||||
|
||||
# Разделение данных на обучающую и тестовую выборки
|
||||
train_data, test_data, train_labels, test_labels = train_test_split(data[features], data[target], test_size=0.2, random_state=42)
|
||||
|
||||
# Создание и обучение модели дерева решений
|
||||
model = DecisionTreeClassifier()
|
||||
model.fit(train_data, train_labels)
|
||||
|
||||
# Прогнозирование на тестовой выборке
|
||||
predictions = model.predict(test_data)
|
||||
|
||||
# Оценка точности модели
|
||||
accuracy = accuracy_score(test_labels, predictions)
|
||||
print('Точность модели на тестовой выборке:', accuracy)
|
||||
|
||||
# Прогнозирование на оставшемся проценте данных
|
||||
remaining_data = data.drop(test_data.index)
|
||||
remaining_predictions = model.predict(remaining_data[features])
|
||||
|
||||
# Вывод результатов
|
||||
print('Прогнозы по оставшемуся проценту данных:', remaining_predictions)
|
||||
|
||||
# Сделайте необходимые выводы
|
||||
1
kutygin_andre_lab_3/ufo_sighting_data.csv
Normal file
1
kutygin_andre_lab_3/ufo_sighting_data.csv
Normal file
File diff suppressed because one or more lines are too long
108
mashkova_margarita_lab_4/README.md
Normal file
108
mashkova_margarita_lab_4/README.md
Normal file
@@ -0,0 +1,108 @@
|
||||
# Лабораторная работа №4
|
||||
## ПИбд-42 Машкова Маргарита (Вариант 19)
|
||||
## Задание
|
||||
C помощью метод кластеризации DBSCAN решить задачу:
|
||||
Сгруппировать телефоны по объему аккумулятора, размеру экрана, RAM и прочим показателям.
|
||||
Интерпретировать результаты и оценить, насколько хорошо он подходит для решения сформулированной задачи.
|
||||
|
||||
### Данные:
|
||||
> Датасет о характеристиках мобильных телефонов и их ценах
|
||||
>
|
||||
> Ссылка на датасет в kaggle: [Mobile Phone Specifications and Prices](https://www.kaggle.com/datasets/pratikgarai/mobile-phone-specifications-and-prices/data)
|
||||
|
||||
### Модели:
|
||||
|
||||
> - DBSCAN
|
||||
|
||||
> **Note**
|
||||
>
|
||||
> Метод DBSCAN позволяет получить оптимальное разбиение точек на оптимальное количество кластеров,
|
||||
а также выявить атипичные объекты (шум). Оценивается при помощи метрики `Силуэт (Silhouette Score)` - насколько
|
||||
плотно объекты прилегают к центру кластера. Данная метрика оценивает качество кластеризации путем
|
||||
измерения среднего значения силуэта для каждой точки данных.
|
||||
Значение Silhouette Score находится в диапазоне от -1 до 1, где ближе к 1 означает лучшую кластеризацию.
|
||||
Высокое значение Silhouette Score указывает на хорошую разделимость кластеров,
|
||||
а низкое значение может указывать на перекрывающиеся кластеры.
|
||||
|
||||
## Запуск программы
|
||||
Для запуска программы необходимо запустить файл main.py
|
||||
|
||||
## Используемые технологии
|
||||
> **Язык программирования:** python
|
||||
>
|
||||
> **Библиотеки:**
|
||||
> - `pandas` - предоставляет функциональность для обработки и анализа набора данных.
|
||||
> - `sklearn` - предоставляет широкий спектр инструментов для машинного обучения, статистики и анализа данных.
|
||||
## Описание работы программы
|
||||
|
||||
### Описание набора данных
|
||||
Данный набор содержит характеристики различных телефонов, в том числе их цену.
|
||||
|
||||
Названия столбцов набора данных и их описание:
|
||||
|
||||
- **Id** - идентификатор строки (int)
|
||||
- **Name** - наименование телефона (string)
|
||||
- **Brand** - наименование бренда телефона (string)
|
||||
- **Model** - модель телефона (string)
|
||||
- **Battery capacity (mAh)** - емкость аккумулятора в мАч (int)
|
||||
- **Screen size (inches)** - размер экрана в дюймах по противоположным углам (float)
|
||||
- **Touchscreen** - имеет телефон сенсорный экран или нет (string - Yes/No)
|
||||
- **Resolution x** - разрешение телефона по ширине экрана (int)
|
||||
- **Resolution y** - разрешение телефона по высоте экрана (int)
|
||||
- **Processor** - количество ядер процессора (int)
|
||||
- **RAM (MB)** - доступная оперативная память телефона в МБ (int)
|
||||
- **Internal storage (GB)** - внутренняя память телефона в ГБ (float)
|
||||
- **Rear camera** - разрешение задней камеры в МП (0, если недоступно) (float)
|
||||
- **Front camera** - разрешение фронтальной камеры в МП (0, если недоступно) (float)
|
||||
- **Operating system** - ОС, используемая в телефоне (string)
|
||||
- **Wi-Fi** - имеет ли телефон функция Wi-Fi (string - Yes/No)
|
||||
- **Bluetooth** - имеет ли телефон функцию Bluetooth (string - Yes/No)
|
||||
- **GPS** - имеет ли телефон функцию GPS (string - Yes/No)
|
||||
- **Number of SIMs** - количество слотов для SIM-карт в телефоне (int)
|
||||
- **3G** - имеет ли телефон сетевую функкцию 3G (string - Yes/No)
|
||||
- **4G/ LTE** - имеет ли телефон сетевую функкцию 4G/LTE (string - Yes/No)
|
||||
- **Price** - цена телефона в индийских рупиях (int)
|
||||
|
||||
Первоначально данные обрабатываются: все строковые значения признаков необходимо привести к численным.
|
||||
|
||||
Метод не требует предварительных предположений о числе кластеров,
|
||||
но нужно настроить два других параметра: `eps` и `min_samples`.
|
||||
Данные параметры – это соответственно максимальное расстояние между соседними точками и минимальное число точек
|
||||
в окрестности (количество соседей), когда можно говорить, что эти экземпляры данных образуют один кластер.
|
||||
|
||||
Далее выполняется два этапа предобработки данных:
|
||||
1. Масшатабирование: Создается экземпляр класса `StandardScaler()`.
|
||||
Метод `fit_transform(data)` этого класса применяется к данным, чтобы выполнить масштабирование.
|
||||
Результат масштабирования сохраняется в переменной `X_scaled`.
|
||||
2. Нормализация: Метод `normalize(X_scaled)` применяется к масштабированным данным `X_scaled`, чтобы выполнить нормализацию.
|
||||
Результат нормализации сохраняется в переменной `X_normalized`.
|
||||
|
||||
Затем выполняется анализ главных компонент (PCA) на нормализованных данных и преобразование их в двумерное пространство.
|
||||
Это позволяет снизить размерность данных и выделить наиболее информативные признаки.
|
||||
Результат преобразования сохраняется в переменной `X_principal`.
|
||||
|
||||
Создается и обучается модель `dbscan` на нормализованных и масштабированных данных.
|
||||
На основе всех признаков данные разбиваются на кластеры. В поле `labels` в алгоритме DBSCAN хранятся метки кластеров,
|
||||
к которым были присвоены точки данных.
|
||||
Каждая точка данных будет иметь свою метку, указывающую на принадлежность к определенному кластеру или на то,
|
||||
что она не принадлежит ни одному кластеру и считается выбросом (кластер `-1`).
|
||||
|
||||
Вычисляется и выводятся в консоль количество кластеров и шумовых точек.
|
||||
|
||||
Качество кластеризации оценивается с помощью `silhouette_score` и выводится в консоль.
|
||||
|
||||
Отображается график кластеризации.
|
||||
|
||||
## Тесты
|
||||
### Получившиеся кластеры и оценка качества кластеризации:
|
||||
|
||||

|
||||
|
||||
### График кластеризации:
|
||||
|
||||

|
||||
|
||||
Шумовые точки изображены фиолетовым цветом.
|
||||
|
||||
**Вывод:** исходя из полученных результатов, значение метрики "силуэт" составляет всего 2%, что указывает на низкое
|
||||
качество кластеризации и перекрывающиеся кластеры. Метод кластеризации `DBSCAN` плохо работает на этих данных.
|
||||
BIN
mashkova_margarita_lab_4/clusters.png
Normal file
BIN
mashkova_margarita_lab_4/clusters.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 21 KiB |
BIN
mashkova_margarita_lab_4/dbscan_plot.png
Normal file
BIN
mashkova_margarita_lab_4/dbscan_plot.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 112 KiB |
72
mashkova_margarita_lab_4/main.py
Normal file
72
mashkova_margarita_lab_4/main.py
Normal file
@@ -0,0 +1,72 @@
|
||||
import pandas as pd
|
||||
import matplotlib.pyplot as plt
|
||||
from sklearn.feature_extraction.text import TfidfVectorizer
|
||||
from sklearn.preprocessing import LabelEncoder
|
||||
from sklearn.cluster import DBSCAN
|
||||
from sklearn.decomposition import PCA
|
||||
from sklearn.metrics import silhouette_score
|
||||
from sklearn.preprocessing import StandardScaler
|
||||
from sklearn.preprocessing import normalize
|
||||
from collections import Counter
|
||||
|
||||
filename = "mobiles.csv"
|
||||
# Считываем данные из файла в DataFrame
|
||||
data = pd.read_csv(filename, sep=',')
|
||||
# Удаляем столбец с идентификатором
|
||||
data.pop("Id")
|
||||
|
||||
# Приведение строковых значений признаков к численным при помощи векторайзера с суммированием
|
||||
FEATURE_COLUMNS_TO_PROC = ['Name', 'Brand', 'Model', 'Operating system']
|
||||
for column_name in FEATURE_COLUMNS_TO_PROC:
|
||||
vectorizer = TfidfVectorizer()
|
||||
train_text_feature_matrix = vectorizer.fit_transform(data[column_name]).toarray()
|
||||
a = pd.DataFrame(train_text_feature_matrix)
|
||||
data[column_name] = a[a.columns[1:]].apply(lambda x: sum(x.dropna().astype(float)), axis=1)
|
||||
|
||||
# Приведение строковых значений к численным при помощи числового кодирования LabelEncoder
|
||||
le = LabelEncoder()
|
||||
data['Touchscreen'] = le.fit_transform(data['Touchscreen'])
|
||||
data['Wi-Fi'] = le.fit_transform(data['Wi-Fi'])
|
||||
data['Bluetooth'] = le.fit_transform(data['Bluetooth'])
|
||||
data['GPS'] = le.fit_transform(data['GPS'])
|
||||
data['3G'] = le.fit_transform(data['3G'])
|
||||
data['4G/ LTE'] = le.fit_transform(data['4G/ LTE'])
|
||||
|
||||
scaler = StandardScaler()
|
||||
X_scaled = scaler.fit_transform(data)
|
||||
# Нормализация масштабированных данных
|
||||
X_normalized = normalize(X_scaled)
|
||||
X_normalized = pd.DataFrame(X_normalized)
|
||||
|
||||
pca = PCA(n_components=2)
|
||||
X_principal = pca.fit_transform(X_normalized)
|
||||
X_principal = pd.DataFrame(X_principal)
|
||||
X_principal.columns = ['P1', 'P2']
|
||||
|
||||
# Определяем и обучаем модель
|
||||
dbscan = DBSCAN(eps=0.05, min_samples=5).fit(X_principal)
|
||||
labels = dbscan.labels_
|
||||
|
||||
# Вычисление количества кластеров
|
||||
N_clus = len(set(labels))-(1 if -1 in labels else 0)
|
||||
print('Количество получившихся кластеров: %d' % N_clus)
|
||||
counter = Counter(dbscan.labels_)
|
||||
clusters_df = pd.DataFrame({'Номер кластера': counter.keys(), 'Кол-во элементов': counter.values()}) \
|
||||
.sort_values(by='Кол-во элементов', ascending=False)
|
||||
print(clusters_df.reset_index(drop=True))
|
||||
|
||||
# Вычисление количества шумовых точек
|
||||
n_noise = list(dbscan.labels_).count(-1)
|
||||
print('Количество шумовых точек: %d' % n_noise)
|
||||
|
||||
# Оценка качества кластеризации с помощью silhouette_score
|
||||
silhouette_avg = silhouette_score(X_principal, labels)
|
||||
|
||||
print("Silhouette Score:", silhouette_avg)
|
||||
|
||||
plt.scatter(X_principal['P1'], X_principal['P2'], c=labels)
|
||||
plt.savefig('dbscan_plot.png')
|
||||
# plt.show()
|
||||
|
||||
|
||||
|
||||
1360
mashkova_margarita_lab_4/mobiles.csv
Normal file
1360
mashkova_margarita_lab_4/mobiles.csv
Normal file
File diff suppressed because it is too large
Load Diff
Reference in New Issue
Block a user