用类封装起来,以后使用起来将方便多了。
import numpy as np
from sklearn import preprocessingfrom 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