Sklearn相關

一、添加權重

clf = RandomForestClassifier(n_estimators=10,class_weight ={0:0.81,1:0.19})

二、輸出

pred = clf.predict_proba(test)#爲機率

pred = clf.predict(test)#爲結果

 三、結果集分佈

group_df = train.標籤.value_counts().reset_index()
k = group_df['標籤'].sum()
print((group_df.標籤/k))