probly.transformation.dropout¶

probly.transformation.dropout(base, p=0.25, rng_collection='dropout', rngs=1)[source]¶

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

Parameters:

base (T) – Predictor, The base model to be used for dropout.
p (float) – float, The probability of dropping out a neuron. Default is 0.25.
rng_collection (str) – Optional str for flax layer initialization. Default is “dropout”.
rngs (Rngs | RngStream | int) – Optional rngs for flax layer initialization (types: rnglib.Rngs | rnglib.RngStream | int), default: 1.

Returns:

Predictor, The DropOut predictor.

Return type: