[ Foro de Python ]

machine learning

Tienen que aumentar un parametro para la profundid en la definicion de la funcion evalRFC. Esta permite cambiar el valor cada vez cuando estan ejecutando la funcion, pero no logro que corra
def evalRFC(n_trees=10):
   # create the perceptron
   rfc = RandomForestClassifier(n_estimators=n_trees, max_depth=3)
   #train the perceptron
   rfc.fit(x_train, y_train)
   # evaluate errors
   acc = rfc.score(x_test, y_test)
   return acc

depth = [1,2,3]
# a esta lista pueden aumentar las resultados de cada profundidad
results = []

# tienen que utilizar el parametro d para controlar la profundidad...
for d in depth:
   acc = []
   for i in range(1,50):
   
       acc.append(evalRFC(n_trees=i))

   
plt.figure()
plt.plot(np.asarray(results).T)
plt.legend(['Depth 1','Depth 2','Depth 3'])
plt.xlabel('# Trees')
plt.ylabel('Precision')
plt.show