Class: ID3

ID3()

La clase ID3 crea un árbol de decisión.

Constructor

new ID3()

Source:

Members

tree :object

Type:
  • object
Source:

Methods

buildTree(data, features) → {object}

Construye el árbol de decisión usando el algoritmo ID3.
Parameters:
Name Type Description
data Array Representa cada una de las instancias del conjunto de datos.
features Array Representa los atributos de las instancias.
Source:
Returns:
Type
object

calculateEntropy(classes) → {object}

Calcula la entropía de un conjunto de clases.
Parameters:
Name Type Description
classes Array Representa las clases del conjunto de datos.
Source:
Returns:
Type
object

calculateEntropyForFeature(values, data) → {object}

Calcula la entropía condicional para una característica específica.
Parameters:
Name Type Description
values Array Representa un conjunto de atributos.
data Array Representa todas las instancias del conjunto de datos.
Source:
Returns:
Type
object

chooseBestFeature(data, features) → {Array}

Elige la mejor característica para dividir el conjunto de datos.
Parameters:
Name Type Description
data Array Representa cada una de las instancias del conjunto de datos.
features Array Representa los atributos de las instancias.
Source:
Returns:
Type
Array

getClassifierName() → {string}

Regresa el nombre del modelo usado.
Source:
Returns:
Type
string

majorityClass(classes) → {Array}

Calcula la clase mayoritaria.
Parameters:
Name Type Description
classes Array Representa las clases del conjunto de datos.
Source:
Returns:
Type
Array

predict(X) → {Array}

Predice las clases de un nuevo conjunto de datos de características (X).
Parameters:
Name Type Description
X Array Datos para predicción.
Source:
Returns:
Type
Array

predictInstance(instance) → {Array}

Método auxiliar para predecir la clase de una única instancia usando el árbol.
Parameters:
Name Type Description
instance Array Datos para predicción.
Source:
Returns:
Type
Array

train(X, y)

Entrena el modelo con un conjunto de datos de características (X) y etiquetas de clase (y).
Parameters:
Name Type Description
X Array Representa las características de las instancias.
y Array Representa las etiquetas de clase correspondientes.
Source: