ToB企服应用市场:ToB评测及商务社交产业平台

标题: 9. 呆板学习汇总(数据、模型、流程、心血管疾病预测) [打印本页]

作者: 花瓣小跑    时间: 2024-7-24 17:46
标题: 9. 呆板学习汇总(数据、模型、流程、心血管疾病预测)
1. 数据

  
  2. 模型

  
  3. 流程

    4. 代码(预测心血管疾病)

  4.0 读入数据,数据规范化处理

  1. from sklearn.datasets import make_classification
  2. from sklearn.model_selection import train_test_split
  3. import numpy as np
  4. import time
  5. # 读取数据
  6. X=[]
  7. y=[]
  8. file_name='./心血管疾病数据集.csv'
  9. with open(file = file_name,mode='r',encoding='utf8') as f:
  10.     line_first = np.array(f.readline().strip().split(','))
  11.     for line in f:  
  12.         if line:
  13.             line = f.readline().strip().split(',')
  14.             X.append(line[:-1])
  15.             y.append(line[-1])
  16. # 将数据转为numpy数组
  17. X=np.array(X)
  18. y=np.array(y)
  19. # 把所有数据都转为float类型
  20. X=X.astype(float)
  21. y=y.astype(float)
  22. # 删掉第1列id
  23. X=X[:,1:]
  24. # 切分数据
  25. from sklearn.model_selection import train_test_split
  26. X_train,X_test,y_train,y_test = train_test_split(X,y,test_size=0.2)
  27. # 数据预处理(规范化)
  28. _mean = X_train.mean(axis=0)
  29. _std = X_train.std(axis=0)
  30. X_train = (X_train-_mean)/(_std+1e-9)
  31. X_test = (X_test-_mean)/(_std+1e-9)
复制代码
4.1 KNN

  1. """
  2.     测试1:KNN
  3. """
  4. from sklearn.neighbors import KNeighborsClassifier
  5. knn = KNeighborsClassifier(n_neighbors=5)
  6. # 取一个时间戳
  7. start_fit = time.time()
  8. knn.fit(X=X_train, y=y_train)
  9. # 取一个时间戳
  10. start_predict = time.time()
  11. y_pred = knn.predict(X=X_test)
  12. # 取一个时间戳
  13. stop_predict = time.time()
  14. # 评估
  15. acc = ( y_pred== y_test).mean()
  16. # 打印结果
  17. print(f"""KNN:
  18. --> 训练耗时:{start_predict-start_fit} 秒;
  19. --> 推理耗时:{stop_predict-start_predict} 秒;
  20. --> 准确率:{acc} ;""")
复制代码
KNN:
–> 训练耗时:0.04751253128051758 秒;
–> 推理耗时:1.3453574180603027 秒;
–> 正确率:0.6402857142857142 ;
  4.2 高斯朴素贝叶斯

  1. """
  2.     测试2:高斯朴素贝叶斯
  3. """
  4. from sklearn.naive_bayes import GaussianNB
  5. gnb = GaussianNB()
  6. # 取一个时间戳
  7. start_fit = time.time()
  8. gnb.fit(X=X_train, y=y_train)
  9. # 取一个时间戳
  10. start_predict = time.time()
  11. y_pred = gnb.predict(X=X_test)
  12. # 取一个时间戳
  13. stop_predict = time.time()
  14. # 评估
  15. acc = ( y_pred== y_test).mean()
  16. # 打印结果
  17. print(f"""GNB:
  18. --> 训练耗时:{start_predict-start_fit} 秒;
  19. --> 推理耗时:{stop_predict-start_predict} 秒;
  20. --> 准确率:{acc} ;""")
复制代码
GNB:
–> 训练耗时:0.009502649307250977 秒;
–> 推理耗时:0.0010018348693847656 秒;
–> 正确率:0.5931428571428572 ;
  4.3 决策树

  1. """
  2.     测试3:决策树
  3. """
  4. from sklearn.tree import DecisionTreeClassifier
  5. dtc = DecisionTreeClassifier()
  6. # 取一个时间戳
  7. start_fit = time.time()
  8. dtc.fit(X=X_train, y=y_train)
  9. # 取一个时间戳
  10. start_predict = time.time()
  11. y_pred = dtc.predict(X=X_test)
  12. # 取一个时间戳
  13. stop_predict = time.time()
  14. # 评估
  15. acc = ( y_pred==y_test).mean()
  16. # 打印结果
  17. print(f"""DTC:
  18. --> 训练耗时:{start_predict-start_fit} 秒;
  19. --> 推理耗时:{stop_predict-start_predict} 秒;
  20. --> 准确率:{acc} ;""")
复制代码
DTC:
–> 训练耗时:0.11947154998779297 秒;
–> 推理耗时:0.002510547637939453 秒;
–> 正确率:0.6367142857142857 ;
  4.4 随机丛林

  1. """
  2.     测试4:随机森林
  3. """
  4. from sklearn.ensemble import RandomForestClassifier
  5. rfc = RandomForestClassifier()
  6. # 取一个时间戳
  7. start_fit = time.time()
  8. rfc.fit(X=X_train, y=y_train)
  9. # 取一个时间戳
  10. start_predict = time.time()
  11. y_pred = rfc.predict(X=X_test)
  12. # 取一个时间戳
  13. stop_predict = time.time()
  14. # 评估
  15. acc = ( y_pred==y_test).mean()
  16. # 打印结果
  17. print(f"""RFC:
  18. --> 训练耗时:{start_predict-start_fit} 秒;
  19. --> 推理耗时:{stop_predict-start_predict} 秒;
  20. --> 准确率:{acc} ;""")
复制代码
RFC:
–> 训练耗时:3.7813894748687744 秒;
–> 推理耗时:0.18322372436523438 秒;
–> 正确率:0.715 ;
  4.5 支持向量机

  1. """
  2.     测试5:支持向量机
  3. """
  4. from sklearn.svm import SVC
  5. svc = SVC()
  6. # 取一个时间戳
  7. start_fit = time.time()
  8. svc.fit(X=X_train, y=y_train)
  9. # 取一个时间戳
  10. start_predict = time.time()
  11. y_pred = svc.predict(X=X_test)
  12. # 取一个时间戳
  13. stop_predict = time.time()
  14. # 评估
  15. acc = ( y_pred==y_test).mean()
  16. # 打印结果
  17. print(f"""SVC:
  18. --> 训练耗时:{start_predict-start_fit} 秒;
  19. --> 推理耗时:{stop_predict-start_predict} 秒;
  20. --> 准确率:{acc} ;""")
复制代码
SVC:
–> 训练耗时:22.884344339370728 秒;
–> 推理耗时:10.314218997955322 秒;
–> 正确率:0.7188571428571429 ;

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作! 更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。




欢迎光临 ToB企服应用市场:ToB评测及商务社交产业平台 (https://dis.qidao123.com/) Powered by Discuz! X3.4