faskhutdinov_idris_lab_2 is ready
This commit is contained in:
parent
039381211a
commit
2cc422ae74
46023
faskhutdinov_idris_lab_3/Clean_Data_pakwheels.csv
Normal file
46023
faskhutdinov_idris_lab_3/Clean_Data_pakwheels.csv
Normal file
File diff suppressed because it is too large
Load Diff
84
faskhutdinov_idris_lab_3/Readme.md
Normal file
84
faskhutdinov_idris_lab_3/Readme.md
Normal file
@ -0,0 +1,84 @@
|
|||||||
|
# Лабораторная работа №3. Деревья решений
|
||||||
|
## 6 вариант
|
||||||
|
### Задание:
|
||||||
|
Решите с помощью библиотечной реализации дерева решений
|
||||||
|
задачу из лабораторной работы «Веб-сервис «Дерево решений» по предмету
|
||||||
|
«Методы искусственного интеллекта» на 99% ваших данных. Проверьте
|
||||||
|
работу модели на оставшемся проценте, сделайте вывод
|
||||||
|
|
||||||
|
В моем случае данными является датасет о продаже автомобилей. В датасете представлены следующие столбцы:
|
||||||
|
* id
|
||||||
|
* Company Name
|
||||||
|
* Model Name
|
||||||
|
* Price
|
||||||
|
* Model Year
|
||||||
|
* Location
|
||||||
|
* Mileage
|
||||||
|
* Engine Type
|
||||||
|
* Engine Capacity
|
||||||
|
* Color
|
||||||
|
* Assembly
|
||||||
|
* Body Type
|
||||||
|
* Transmission Type
|
||||||
|
* Registration Status
|
||||||
|
|
||||||
|
### Как запустить лабораторную
|
||||||
|
1. Запустить файл main.py
|
||||||
|
|
||||||
|
### Используемые технологии
|
||||||
|
1. Библиотека pandas
|
||||||
|
2. Библиотека scikit-learn
|
||||||
|
3. Python
|
||||||
|
4. IDE PyCharm
|
||||||
|
|
||||||
|
### Описание лабораторной работы
|
||||||
|
|
||||||
|
Программа загружает данные из файла Clean_Data_pakwheels.csv, после чего выбирает необходимые для создания модели столбцы.
|
||||||
|
Выбранные столбцы разделяются на целевую переменную (Y) и признаки (X). Некоторые столбцы в датасете представлены в виде текстовых значений, поэтому мы представляем их как численные значения
|
||||||
|
Затем программа обучает модель, выполняет прогнозы и оценивает точность. В консоль выводятся признаки по их важности
|
||||||
|
Целевой признак - Registration Status
|
||||||
|
### Результат
|
||||||
|
Accuracy: 0.9327548806941431
|
||||||
|
* Признак Важность
|
||||||
|
* 1 Mileage 0.332722
|
||||||
|
* 2 Price 0.332358
|
||||||
|
* 0 Model Year 0.175522
|
||||||
|
* 34 Transmission Type_Automatic 0.086699
|
||||||
|
* 13 Company Name_Honda 0.021243
|
||||||
|
* 31 Company Name_Toyota 0.015743
|
||||||
|
* 30 Company Name_Suzuki 0.008819
|
||||||
|
* 10 Company Name_Daihatsu 0.007749
|
||||||
|
* 25 Company Name_Nissan 0.007616
|
||||||
|
* 4 Company Name_Audi 0.003018
|
||||||
|
* 23 Company Name_Mercedes 0.001886
|
||||||
|
* 22 Company Name_Mazda 0.001800
|
||||||
|
* 18 Company Name_KIA 0.001416
|
||||||
|
* 24 Company Name_Mitsubishi 0.001044
|
||||||
|
* 29 Company Name_Subaru 0.000787
|
||||||
|
* 5 Company Name_BMW 0.000458
|
||||||
|
* 19 Company Name_Land 0.000407
|
||||||
|
* 27 Company Name_Range 0.000332
|
||||||
|
* 26 Company Name_Porsche 0.000331
|
||||||
|
* 35 Transmission Type_Manual 0.000050
|
||||||
|
* 20 Company Name_Lexus 0.000000
|
||||||
|
* 21 Company Name_MINI 0.000000
|
||||||
|
* 9 Company Name_Daewoo 0.000000
|
||||||
|
* 8 Company Name_DFSK 0.000000
|
||||||
|
* 14 Company Name_Hummer 0.000000
|
||||||
|
* 7 Company Name_Chevrolet 0.000000
|
||||||
|
* 11 Company Name_FAW 0.000000
|
||||||
|
* 17 Company Name_Jeep 0.000000
|
||||||
|
* 28 Company Name_SsangYong 0.000000
|
||||||
|
* 16 Company Name_Jaguar 0.000000
|
||||||
|
* 6 Company Name_Chery 0.000000
|
||||||
|
* 15 Company Name_Hyundai 0.000000
|
||||||
|
* 32 Company Name_United 0.000000
|
||||||
|
* 33 Company Name_Volvo 0.000000
|
||||||
|
* 3 Company Name_Adam 0.000000
|
||||||
|
* 12 Company Name_Fiat 0.000000
|
||||||
|
|
||||||
|
### Вывод
|
||||||
|
Исходя из результатов работы программы можно сделать вывод, что наиболее важным признаком, отвечающим за
|
||||||
|
то, зарегистрирована машина или нет, является её пробег, а так же её цена на рынке. Точность модели составляет 93%, что говорит о том,
|
||||||
|
что она классифицирует данные при заданных условиях с высокой точностью.
|
||||||
|
|
39
faskhutdinov_idris_lab_3/main.py
Normal file
39
faskhutdinov_idris_lab_3/main.py
Normal file
@ -0,0 +1,39 @@
|
|||||||
|
import pandas as pd
|
||||||
|
from sklearn.model_selection import train_test_split
|
||||||
|
from sklearn.tree import DecisionTreeClassifier
|
||||||
|
from sklearn.metrics import accuracy_score
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
# Чтение данных из csv файла
|
||||||
|
data = pd.read_csv("Clean_Data_pakwheels.csv")
|
||||||
|
# Выбор необходимых для создания модели столбцов
|
||||||
|
selected_columns = ['Company Name', 'Model Year', 'Mileage', 'Transmission Type', 'Price', 'Registration Status']
|
||||||
|
data = data[selected_columns]
|
||||||
|
# Разделение данных на признаки (X) и целевую переменную (y), целевая переменная в данном случае Registration Status
|
||||||
|
y = data['Registration Status']
|
||||||
|
data = data.drop(columns=['Registration Status'])
|
||||||
|
# В связи с тем, что некоторые столбцы представляют из себя текстовые значения, мы представляем их в виде числовых значений
|
||||||
|
X = pd.get_dummies(data)
|
||||||
|
# Тестовый набор в данном случае - 1%, обучающий - 99%
|
||||||
|
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.01)
|
||||||
|
model = DecisionTreeClassifier()
|
||||||
|
model.fit(X_train, y_train)
|
||||||
|
# Предсказание на тестовом наборе
|
||||||
|
y_pred = model.predict(X_test)
|
||||||
|
|
||||||
|
# Оценка точности модели
|
||||||
|
accuracy = accuracy_score(y_test, y_pred)
|
||||||
|
print(f"Accuracy: {accuracy}")
|
||||||
|
|
||||||
|
# Важность признаков
|
||||||
|
importance = pd.DataFrame({'Признак': X.columns, 'Важность': model.feature_importances_})
|
||||||
|
importance = importance.sort_values(by='Важность', ascending=False)
|
||||||
|
print(importance)
|
||||||
|
|
||||||
|
|
||||||
|
main()
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user