Shortcuts

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.