Neural Network Models

mllabs.nn.NNClassifier

Bases: _NNBase, ClassifierMixin

predict(X)

predict_proba(X)

mllabs.nn.NNRegressor

Bases: _NNBase, RegressorMixin

predict(X)

mllabs.nn.SimpleConcatHead

Bases: NNHead

mllabs.nn.FTTransformerHead

Bases: NNHead

mllabs.nn.DenseHidden

Bases: NNHidden

mllabs.nn.LogitOutput

Bases: NNOutput

mllabs.nn.BinaryLogitOutput

Bases: NNOutput

mllabs.nn.RegressionOutput

Bases: NNOutput

mllabs.adapter.NNAdapter

Bases: ModelAdapter

get_fit_params(data_dict, params=None, logger=None)