python實現(xiàn)H2O中的隨機森林算法介紹及其項目實戰(zhàn)
H2O中的隨機森林算法介紹及其項目實戰(zhàn)(python實現(xiàn))
包的引入:from h2o.estimators.random_forest import H2ORandomForestEstimator
H2ORandomForestEstimator 的常用方法和參數(shù)介紹:
(一)建模方法:
model =H2ORandomForestEstimator(ntrees=n,max_depth =m) model.train(x=random_pv.names,y='Catrgory',training_frame=trainData)
通過trainData來構(gòu)建隨機森林模型,model.train中的trainData:訓(xùn)練集,x:預(yù)測變量名稱,y:預(yù)測 響應(yīng)變量的名稱
(二)預(yù)測方法:
pre_tag=H2ORandomForestEstimator.predict(model ,test_data) 利用訓(xùn)練好的模型來對測試集進行預(yù)測,其中的model:訓(xùn)練好的模型, test_data:測試集。
(三)算法參數(shù)說明:
(1)ntrees:構(gòu)建模型時要生成的樹的棵樹。
(2)max_depth :每棵樹的最大深度。
項目要求:
題目一: 利用train.csv中的數(shù)據(jù),通過H2O框架中的隨機森林算法構(gòu)建分類模型,然后利用模型對 test.csv中的數(shù)據(jù)進行預(yù)測,并計算分類的準(zhǔn)確度進而評價模型的分類效果;通過調(diào)節(jié)參 數(shù),觀察分類準(zhǔn)確度的變化情況。 注:準(zhǔn)確度=預(yù)測正確的數(shù)占樣本數(shù)的比例
題目二: 通過H2o Flow 的隨機森林算法,用同題目一中所用同樣的訓(xùn)練數(shù)據(jù)和參數(shù),構(gòu)建模型; 參看模型中特征的重要性程度,從中選取前8個特征,再去訓(xùn)練模型,并重新預(yù)測結(jié)果, 進而計算分類的準(zhǔn)確度。
需求完成內(nèi)容:2個題目的代碼,認為最好的準(zhǔn)確度的輸出值和test數(shù)據(jù)與預(yù)測結(jié)果合并 后的數(shù)據(jù)集,命名為predict.csv
python實現(xiàn)代碼如下:
(1) 題目一:
#手動進行調(diào)節(jié)參數(shù)得到最好的準(zhǔn)確率
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import h2o
h2o.init()
from h2o.estimators.random_forest import H2ORandomForestEstimator
from __future__ import division
df=h2o.import_file('train.csv')
trainData=df[2:]
model=H2ORandomForestEstimator(ntrees=6,max_depth =16)
model.train(x=trainData.names,y='Catrgory',training_frame=trainData)
df2=h2o.import_file('test.csv')
test_data=df2[2:]
pre_tag=H2ORandomForestEstimator.predict(model ,test_data)
predict=df2.concat(pre_tag)
dfnew=predict[predict['Catrgory']==predict['predict']]
Precision=dfnew.nrow/predict.nrow
print(Precision)
h2o.download_csv(predict,'predict.csv')
運行結(jié)果最好為87.0833%-6-16,如下

#for循環(huán)進行調(diào)節(jié)參數(shù)得到最好的準(zhǔn)確率
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import h2o
h2o.init()
from h2o.estimators.random_forest import H2ORandomForestEstimator
from __future__ import division
df=h2o.import_file('train.csv')
trainData=df[2:]
df2=h2o.import_file('test.csv')
test_data=df2[2:]
Precision=0
nt=0
md=0
for i in range(1,50):
for j in range(1,50):
model=H2ORandomForestEstimator(ntrees=i,max_depth =j)
model.train(x=trainData.names,y='Catrgory',training_frame=trainData)
pre_tag=H2ORandomForestEstimator.predict(model ,test_data)
predict=df2.concat(pre_tag)
dfnew=predict[predict['Catrgory']==predict['predict']]
p=dfnew.nrow/predict.nrow
if Precision<p:
Precision=p
nt=i
md=j
print(Precision)
print(i)
print(j)
h2o.download_csv(predict,'predict.csv')
運行結(jié)果最好為87.5%-49-49,如下

(2)題目二:建模如下,之后挑出排名前8的特征進行再次建模

#手動調(diào)節(jié)參數(shù)得到最大準(zhǔn)確率
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import h2o
h2o.init()
from h2o.estimators.random_forest import H2ORandomForestEstimator
from __future__ import division
df=h2o.import_file('train.csv')
trainData=df[['Average_speed','r_a','r_b','v_a','v_d','Average_RPM','Variance_speed','v_c','Catrgory']]
df2=h2o.import_file('test.csv')
test_data=df2[['Average_speed','r_a','r_b','v_a','v_d','Average_RPM','Variance_speed','v_c','Catrgory']]
model=H2ORandomForestEstimator(ntrees=5,max_depth =18)
model.train(x=trainData.names,y='Catrgory',training_frame=trainData)
pre_tag=H2ORandomForestEstimator.predict(model ,test_data)
predict=df2.concat(pre_tag)
dfnew=predict[predict['Catrgory']==predict['predict']]
Precision=dfnew.nrow/predict.nrow
print(Precision)
h2o.download_csv(predict,'predict.csv')
運行結(jié)果最好為87.5%-5-18,如下

#for循環(huán)調(diào)節(jié)參數(shù)得到最大正確率
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import h2o
h2o.init()
from h2o.estimators.random_forest import H2ORandomForestEstimator
from __future__ import division
df=h2o.import_file('train.csv')
trainData=df[['Average_speed','r_a','r_b','v_a','v_d','Average_RPM','Variance_speed','v_c','Catrgory']]
df2=h2o.import_file('test.csv')
test_data=df2[['Average_speed','r_a','r_b','v_a','v_d','Average_RPM','Variance_speed','v_c','Catrgory']]
Precision=0
nt=0
md=0
for i in range(1,50):
for j in range(1,50):
model=H2ORandomForestEstimator(ntrees=i,max_depth =j)
model.train(x=trainData.names,y='Catrgory',training_frame=trainData)
pre_tag=H2ORandomForestEstimator.predict(model ,test_data)
predict=df2.concat(pre_tag)
dfnew=predict[predict['Catrgory']==predict['predict']]
p=dfnew.nrow/predict.nrow
if Precision<p:
Precision=p
nt=i
md=j
print(Precision)
print(i)
print(j)
h2o.download_csv(predict,'predict.csv')
運行結(jié)果最好為87.5%-49-49,如下

以上就是本文的全部內(nèi)容,希望對大家的學(xué)習(xí)有所幫助,也希望大家多多支持腳本之家。
相關(guān)文章
Matlab中的mat數(shù)據(jù)轉(zhuǎn)成python中使用的npy數(shù)據(jù)遇到的坑及解決
這篇文章主要介紹了Matlab中的mat數(shù)據(jù)轉(zhuǎn)成python中使用的npy數(shù)據(jù)遇到的坑及解決,具有很好的參考價值,希望對大家有所幫助。如有錯誤或未考慮完全的地方,望不吝賜教2022-12-12

