[ Foro de Python ]

machine learning

11-Jan-2020 18:37
Invitado (Jhi)
0 Respuestas

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




Si ya eres usuario del sistema, puedes contestar desde tu cuenta y así ganar prestigio.

Si sólo eres un visitante, puedes optar por...