Segue abaixo o código:
def Validador(x,y):
validador = StratifiedShuffleSplit(n_splits= 1, test_size=0.1, random_state=0)
for treino_id, teste_id in validador.split(x, y):
x_train, x_test = x[treino_id], x[teste_id]
y_train, y_test = y[treino_id], y[teste_id]
return x_train,x_test,y_train,y_test
def Executar_Classificador(classificador,x_train,x_test,y_train):
arvore = classificador.fit(x_train,y_train)
y_predict = arvore.predict(x_test)
return y_predict,classificador
x = dados.drop(columns='Class').values
y = dados['Class'].values
classificador = tree.DecisionTreeClassifier()
x_train,x_test,y_train,y_test = Validador(x,y)
y_predict = Executar_Classificador(classificador,x_train,x_test,y_train)
print(classificador)
Como posso observar os atributos?
