educoder-随机森林（初）

博主：微风
发布时间：2021 年 05 月 17 日
2809 次浏览
7 条评论
1133字数
分类：机器学习

第一关：随机森林

from sklearn.ensemble import RandomForestClassifier

def data_classification(train_data, train_label, test_data):
    '''
    使用随机森林对训练集数据进行训练，并对测试集数据进行预测，并返回预测结果
    :param train_data: 训练集数据，类型为ndarray
    :param train_label: 训练集标签，类型为ndarrayfff
    :param test_data: 测试集数据，类型为ndarray
    :return: 分类结果
    '''
    #********* Begin *********#
    rfc = RandomForestClassifier()
    rfc.fit(train_data,train_label)
    return rfc.predict(test_data)
    #********* End *********#

第二关：手写数字识别

from sklearn.ensemble import RandomForestClassifier

def digit_predict(train_image, train_label, test_image):
    '''
    实现功能：训练模型并输出预测结果
    :param train_image: 包含多条训练样本的样本集，类型为ndarray,shape为[-1, 8, 8]
    :param train_label: 包含多条训练样本标签的标签集，类型为ndarray
    :param test_image: 包含多条测试样本的测试集，类型为ndarry
    :return: test_image对应的预测标签，类型为ndarray
    '''

    #************* Begin ************#
    rfc = RandomForestClassifier(n_estimators=500)
    rfc.fit(train_image,train_label)
    return rfc.predict(test_image)
    #************* End **************#

最后修改：2021 年 07 月 01 日

如果觉得我的文章对你有用，请随意赞赏

发表评论取消回复
使用cookie技术保留您的个人信息以便您下次快速评论，继续评论表示您已同意该条款

评论 *

私密评论

名称 *

🎲

邮箱 *

地址

7 条评论

小有名气的匿名人士 Google Chrome 90.0.4430.212 Windows 10 中国广东深圳
2021 年 05 月 19 日

不错

回复
1. 刚下飞机的女孩 Firefox 88.0 Windows 10 中国广东深圳
  2021 年 05 月 19 日
  
  @小有名气的匿名人士
  
  嘿嘿
  
  回复
  1. 宋科成 Google Chrome 89.0.4389.82 Windows 10 中国山东青岛
    2021 年 05 月 19 日
    
    @刚下飞机的女孩
    
    真的烦躁的很哦你
    
    回复
111 Microsoft Edge 90.0.818.56 Windows 10 中国广东深圳
2021 年 05 月 18 日

(ฅ´ω`ฅ)

回复
1. 是微风吖 Safari 604.1 iPhone 14.5 中国广东深圳
  2021 年 05 月 18 日
  
  @111
  
  哈哈ヾ(≧∇≦*)ゝ
  
  回复
  1. 向富裕 Google Chrome 90.0.4430.212 Windows 7 中国广东深圳
    2021 年 05 月 20 日
    
    @是微风吖
    
    罗总给你一个家
    
    回复
    
    是微风吖 Google Chrome 90.0.4430.212 Windows 10 中国广东深圳
    2021 年 05 月 20 日
    
    @向富裕
    
    啊哈。( ,,´･ω･)ﾉ"(´っω･｀｡)
    
    回复