generated/torch_uncertainty.baselines.regression.MLPBaseline
Download Notebook
Notebook
View on GitHub
GitHub
MLPBaseline
-
class torch_uncertainty.baselines.regression.MLPBaseline(output_dim, in_features, loss, version, hidden_dims, num_estimators=1, dropout_rate=0.0, alpha=None, gamma=1, distribution=None)[source]
MLP baseline for regression providing support for various versions.