Python-随机森林实例

用类封装起来，以后使用起来将方便多了。

import numpy as np

from sklearn import preprocessing
from sklearn.ensemble import RandomForestClassifier
import matplotlib.pyplot as plt


class MYRandomForestClassifier:
    def __init__(self,Modul=RandomForestClassifier,dataset=np.empty((3,3))):
        self.modul=Modul        
        self.dataset=dataset
    def ReadFile(self,input_file):
        # Reading the data
        X = []; 
        with open(input_file, 'r') as f:
            for line in f.readlines():
#                data=line.split(',')
                data = line[:-1].split(',')                 
                X.append(data)
        X = np.array(X)
        # Convert string data to numerical data
        label_encoder = [] 
        X_encoded = np.empty(X.shape)
        for