aibox.nlp.estimators.classification.mlp_classifier

Classificar baseado na arquitetura MLP.

Classes

MLPClassifier([hidden_layers, epochs, ...])

Classificador MLP.
class aibox.nlp.estimators.classification.mlp_classifier.MLPClassifier(hidden_layers: list[int] = [256, 128, 64], epochs: int = 10, dropout_prob: float | list[float | None] | None = None, learning_rate: float = 0.0001, optim_params: dict = {}, optim: Literal['adam', 'adamw', 'rmsprop', 'adagrad', 'sgd'] = 'adamw', random_state: int | None = None, train_batch_size: int = 64, device: str | None = None)[código-fonte]

Base: MLPEstimator

Classificador MLP.

Para informações sobre a classe acesse MLPEstimator.

Parâmetros:

  • hidden_layers (list[int])

  • epochs (int)

  • dropout_prob (float | list[float | None] | None)

  • learning_rate (float)

  • optim_params (dict)

  • optim (Literal['adam', 'adamw', 'rmsprop', 'adagrad', 'sgd'])

  • random_state (int | None)

  • train_batch_size (int)

  • device (str)

fit(X: list[int] | list[float] | ndarray[int32] | ndarray[float32] | Tensor, y: list[int] | list[float] | ndarray[int32] | ndarray[float32] | Tensor, **kwargs) → None

Realiza o treinamento do estimador.

Parâmetros:
Tipo de retorno:

None

property hyperparameters: dict

Hiper-parâmetros do modelo. Inclui a seed randômica. Estrutura do dicionário varia entre diferentes estimadores.

Retorna:

dicionário de hiper-parâmetros.

property params: dict

Retorna um dicionário com os parâmetros para esse estimador.

Os parâmetros retornados descrevem totalmente o estado do modelo (e,g. pesos de uma rede, superfícies de decisão, estrutura da árvore de decisão, etc).

Retorna:

parâmetros do estimador.

predict(X: list[int] | list[float] | ndarray[int32] | ndarray[float32] | Tensor, **kwargs) → None

Realiza a predição utilizando os parâmetros atuais do modelo.

Parâmetros:

  • X (list[int] | list[float] | ndarray[int32] | ndarray[float32] | Tensor) – dados de entrada com shape (n_samples, n_features).

  • **kwargs – parâmetros extras que podem ser passados para alguns estimadores.

Retorna:

predições com shape (n_samples,).

Tipo de retorno:

None

property random_state: int

Seed randômica utilizada pelo estimador.

Retorna:

seed.