百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 编程字典 > 正文

使用决策树算法识别用户喜欢或不喜欢的歌曲

toyiye 2024-07-06 00:18 19 浏览 0 评论

当我们听音乐时,有时我们倾向于喜欢或不喜欢某些音乐曲目。可能有不同的原因。用户的偏好可能取决于旋律,歌词,歌曲的含义,节奏和流派等。

因此,考虑到上述事实,让我们使用决策树算法识别歌曲的喜欢或不喜欢。

决策树

决策树是监督学习中最受欢迎和最强大的算法之一。与其他监督学习算法不同,决策树可用于分类和回归问题。它是用于机器学习,统计和数据挖掘的预测模型。决策树是树,其中每个节点表示特征(属性),每个链接(分支)表示决策(规则),每个叶象征结果。

使用决策树算法是因为它模仿了人类思维,因此很容易识别数据并使决策树的预测和逻辑清晰易懂。

不需要数据预处理,也不需要任何变量变换,不受异常值的影响。

有不同类型的决策树算法

1. ID3算法(使用熵和信息增益作为指标)

2. C4.5算法(使用增益比作为指标)

3. CART算法(使用基尼指数作为指标)

分类和回归树(CART)算法是一种基于Gini杂质指数作为分裂准则构建决策树的分类算法。基尼指数是衡量随机选择的元素被错误识别的频率的指标。Sklearn支持基尼指数的“基尼”标准,默认情况下; 它需要“Gini”的价值。Scikit-learn使用CART算法的优化版本。

决策树的实现

由于使用了Kaggle网站的数据集Spotify分类数据集(https://www.kaggle.com/geomack/spotifyclassification)。Kaggle是一个预测建模和分析竞赛的平台,统计人员和数据挖掘者在竞争中生成预测和描述公司和用户上传的数据集的最佳模型。

Spotify歌曲属性数据集是包含2017歌曲及其属性的数据集。数据集创建者已从Spotify的API生成数据集。

数据集包含16列:歌曲名称,艺术家姓名,作为标签的目标和13个轨道属性:声学,舞蹈,持续时间,能量,乐器,键,活力,响度,模式,语音,节奏,时间标记,效价,歌曲名称,艺术家姓名和标签作为目标。

决策树的Python代码如下

#Import the necessary python machine learning libraries

import pandas as pd

import numpy as np

from sklearn import tree

from sklearn.tree import DecisionTreeClassifier, export_graphviz

from sklearn.model_selection import train_test_split

from matplotlib import pyplot as plt

from sklearn.metrics import accuracy_score

import seaborn as sns

import graphviz

import pydotplus

import io

import imageio

#Import the data set

data=pd.read_csv(‘data.csv’)

#Split the dataset into training set and test set

train,test=train_test_split(data,test_size=0.15)

classifier=DecisionTreeClassifier(min_samples_split=100)

features=[“danceability”,”loudness”,”valence”,”energy”,”instrumentalness”,”acousticness”,”key”,”speechiness”,”duration_ms”,”liveness”,”mode”,”tempo”,”valence”]

#Traing the Model and Test the Model

X_train=train[features]

y_train=train[“target”]

X_test=test[features]

y_test=test[“target”]

#Decision Tree

dt=classifier.fit(X_train,y_train)

#Visualize the decision tree

def show_tree(tree,features,path):

f=io.StringIO()

export_graphviz(tree,out_file=f,feature_names=features)

pydotplus.graph_from_dot_data(f.getvalue()).write_png(path)

img=imageio.imread(path)

plt.rcParams[“figure.figsize”]=(20,20)

plt.imshow(img)

show_tree(dt, features,”Decision_Tree_Spotify_Songs.png”)

y_pred=classifier.predict(X_test)

score=accuracy_score(y_test,y_pred)*100

print(“Accuracy using the Decision Tree”,round(score,2),”%”)

最后,获得了72.28%的准确度。过度拟合是构建决策树模型的实际问题。该模型存在过度拟合的问题,当算法继续越来越深入以减少训练集错误时会考虑过度拟合,但会导致测试集误差增加。

可以使用两种方法来避免过度拟合:

  • 预修剪 - 通过尽早停止来修剪树木。如果分割的测量值小于阈值,则停止进一步分割子树。停止节点成为一片叶子。
  • 修剪后 - 在完全成长的树中删除给定树的子集,并用叶节点??替换。

决策树是可解释的:可以识别模型使用的过程。并且需要最小的数据准备。根据数据大小,将分割值的最小值用作100。如果最小的分割值是一个较小的值,则生成密集决策树。一般来说,与其他机器学习算法相比,它给出了数据集的预测精度。

相关推荐

为何越来越多的编程语言使用JSON(为什么编程)

JSON是JavascriptObjectNotation的缩写,意思是Javascript对象表示法,是一种易于人类阅读和对编程友好的文本数据传递方法,是JavaScript语言规范定义的一个子...

何时在数据库中使用 JSON(数据库用json格式存储)

在本文中,您将了解何时应考虑将JSON数据类型添加到表中以及何时应避免使用它们。每天?分享?最新?软件?开发?,Devops,敏捷?,测试?以及?项目?管理?最新?,最热门?的?文章?,每天?花?...

MySQL 从零开始:05 数据类型(mysql数据类型有哪些,并举例)

前面的讲解中已经接触到了表的创建,表的创建是对字段的声明,比如:上述语句声明了字段的名称、类型、所占空间、默认值和是否可以为空等信息。其中的int、varchar、char和decimal都...

JSON对象花样进阶(json格式对象)

一、引言在现代Web开发中,JSON(JavaScriptObjectNotation)已经成为数据交换的标准格式。无论是从前端向后端发送数据,还是从后端接收数据,JSON都是不可或缺的一部分。...

深入理解 JSON 和 Form-data(json和formdata提交区别)

在讨论现代网络开发与API设计的语境下,理解客户端和服务器间如何有效且可靠地交换数据变得尤为关键。这里,特别值得关注的是两种主流数据格式:...

JSON 语法(json 语法 priority)

JSON语法是JavaScript语法的子集。JSON语法规则JSON语法是JavaScript对象表示法语法的子集。数据在名称/值对中数据由逗号分隔花括号保存对象方括号保存数组JS...

JSON语法详解(json的语法规则)

JSON语法规则JSON语法是JavaScript对象表示法语法的子集。数据在名称/值对中数据由逗号分隔大括号保存对象中括号保存数组注意:json的key是字符串,且必须是双引号,不能是单引号...

MySQL JSON数据类型操作(mysql的json)

概述mysql自5.7.8版本开始,就支持了json结构的数据存储和查询,这表明了mysql也在不断的学习和增加nosql数据库的有点。但mysql毕竟是关系型数据库,在处理json这种非结构化的数据...

JSON的数据模式(json数据格式示例)

像XML模式一样,JSON数据格式也有Schema,这是一个基于JSON格式的规范。JSON模式也以JSON格式编写。它用于验证JSON数据。JSON模式示例以下代码显示了基本的JSON模式。{"...

前端学习——JSON格式详解(后端json格式)

JSON(JavaScriptObjectNotation)是一种轻量级的数据交换格式。易于人阅读和编写。同时也易于机器解析和生成。它基于JavaScriptProgrammingLa...

什么是 JSON:详解 JSON 及其优势(什么叫json)

现在程序员还有谁不知道JSON吗?无论对于前端还是后端,JSON都是一种常见的数据格式。那么JSON到底是什么呢?JSON的定义...

PostgreSQL JSON 类型:处理结构化数据

PostgreSQL提供JSON类型,以存储结构化数据。JSON是一种开放的数据格式,可用于存储各种类型的值。什么是JSON类型?JSON类型表示JSON(JavaScriptO...

JavaScript:JSON、三种包装类(javascript 包)

JOSN:我们希望可以将一个对象在不同的语言中进行传递,以达到通信的目的,最佳方式就是将一个对象转换为字符串的形式JSON(JavaScriptObjectNotation)-JS的对象表示法...

Python数据分析 只要1分钟 教你玩转JSON 全程干货

Json简介:Json,全名JavaScriptObjectNotation,JSON(JavaScriptObjectNotation(记号、标记))是一种轻量级的数据交换格式。它基于J...

比较一下JSON与XML两种数据格式?(json和xml哪个好)

JSON(JavaScriptObjectNotation)和XML(eXtensibleMarkupLanguage)是在日常开发中比较常用的两种数据格式,它们主要的作用就是用来进行数据的传...

取消回复欢迎 发表评论:

请填写验证码