音乐分类

代码：

import numpy as np
from scipy import fft
from scipy.io import wavfile
from sklearn.linear_model import LogisticRegression
import random
"""
    使用logistic regression处理音乐数据，音乐数据训练样本的获得是使用快速傅里叶变换：
    1.吧训练集扩大到每类100首歌二部之前的10首歌
    2.同时使用logistic回归和KNN作为分类器
    3.引入一些评价标准来比较logictic和KNN在测试集上的表现
"""
"""
n = 40
# hstack数据拼接
# 在模拟X的时候使用了两个正态分布,分别制定各自的均值,方差,生成40个点
X = np.hstack((norm.rvs(loc=2, size=n, scale=2), norm.rvs(loc=8, size=n, scale=3)))
# zeros使得数据点生成40个0,ones使得数据点生成40个1
y = np.hstack((np.zeros(n), np.ones(n)))

"""
general_list = ["classical", "jazz", "country", "pop", "rock", "metal"]
"""
# 处理原始数据，转化为特征文件
def create_fft(g, n):
    read = "d:/xxx/"+g+"/conberted/"+g+"."+str(n).zfill(5)+".au.wav"
    sample_rate, x = wavfile.read(read)
    fft_features = abs(fft(x)[:1000])
    sad = "d:/trainset/"+g+"."+str(n).zfill(5)+".fft"
    np.save(sad, fft_features)


for g in gener_list:
    for n in range(100):
        create_fft(g, n)
"""
# 加载训练集数据，分割训练集和测试集，进行分类器的训练
# 构造训练集
x = []
y = []
for g in general_list:
    for n in range(100):
        read = "D:/AnalyseData学习资源库/人工智能开发【中】/05_分类器项目案例和神经网络算法【尚学堂·百战程序员】/资料/trainset/"+g+"."+str(n).zfill(5)+".fft"+".npy"
        fft_features = np.load(read)
        x.append(fft_features)
        y.append(general_list.index(g))

x = np.array(x)
y = np.array(y)

# 拆分数据为训练集和测试集
randomIndex = random.sample(range(len(y)), int(len(y)*8/10))
trainX = []
trainY = []
testX = []
testY = []

for i in range(len(y)):
    if i in randomIndex:
        trainX.append(x[i])
        trainY.append(y[i])
    else:
        testX.append(x[i])
        testY.append(y[i])

# 使用sklearn来构建和训练两种分类器

# logistic classifier
model = LogisticRegression()
# train
model.fit(trainX, trainY)
# test
predict = model.predict(testX)
print(testY)
print("--------------------------------------------------")
print(predict)
# error
import math
error = 0.0
for i in range(len(testY)):
    if testY[i] != predict[i]:
        error = error + 1
print(error/len(testY))

print("starting read wavfile...")
sample_rate, test = wavfile.read("D:/AnalyseData学习资源库/人工智能开发【中】/05_分类器项目案例和神经网络算法【尚学堂·百战程序员】/资料/trainset/sample/heibao-wudizirong-remix.wav")
testdata_fft_features = abs(fft(test))[:1000]
type_index = model.predict([testdata_fft_features])[0]

print("预测音乐分类为："+general_list[type_index])

结果：