probly.method.dropout

probly.method.dropout(base: T, p: float = 0.25, rng_collection: str = 'dropout', rngs: Rngs | RngStream | int = 1) → T

Create a Dropout predictor from a base predictor based on [GG16a].

Parameters:

• base – Predictor, The base model to be used for dropout.

• p – float, The probability of dropping out a neuron. Default is 0.25.

• rng_collection – Optional str for flax layer initialization. Default is “dropout”.

• rngs – Optional rngs for flax layer initialization (types: rnglib.Rngs | rnglib.RngStream | int), default: 1.

Returns:

Predictor, The DropOut predictor.