2010-02-18 13 views
5

Estoy buscando una implementación de árbol de decisión ID3 en Python o en cualquier otro idioma que tome como validación y un archivo de prueba como entrada y devuelva las predicciones.ID3 Árbol de decisiones con valores numéricos

Encontré this y esto, pero no pude adaptarlos a los valores numéricos, p. Ej. a Iris dataset.

¿Conoces alguna implementación de árbol ID3 que funcione desde la consola o cualquier escrito en Python? O cualquier sugerencia sobre cómo usar this con valores numéricos será increíble.

Respuesta

2

Tengo un algoritmo similar C4.5 escrito en python. Funciona desde la consola. Si estás interesado lo pongo aquí.

Lo siento por una publicación si no necesita esto.

Por cierto, he probado en conjunto de datos del iris :)

Actualizar:

He subido tanto: código y de datos:

Espero que te ayude.

Por cierto, el programa también se puede dibujar un árbol en "PNG" a través de Graphviz

+1

Sí, voy a apreciar si se puede compartir C4.5. –

+0

Gracias por el código C4.5. Necesito calcular la precisión de la clasificación. ¿Cómo puedo obtener las clases de predicción en tu código? Intenté usar classify (tree_root, ds) pero parece que devuelve las mismas etiquetas de clase que el original -o estoy haciendo algo mal :) –

+2

Sí, tienes razón. Las etiquetas son las mismas porque el proceso de aprendizaje se realiza en conjunto de datos de iris completo. Solo excluya algunos de los vectores de datos del archivo de datos (o hágalo en la función 'load_iris') y use vectores excluidos para probar la predicción. – woo

Cuestiones relacionadas