probly.method.dropconnect

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

Create a DropConnect predictor from a base predictor based on [MNM+19].

Parameters:

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

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

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

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

Returns:

The DropConnect predictor.