code: def init_classification_bias(
      bias: jnp.ndarray, prior_prob: float
  ) -> jnp.ndarray:
      return jnp.full(
          bias.shape, np.log(prior_prob) - np.log1p(-prior_prob)
      )
  
imports: from typing import Any, Dict, List, Optional, Sequence, Tuple, Type, Union
  from absl import logging
  from clu import preprocess_spec
  from flax import core
  import jax
  import jax.numpy
  import numpy
  from scenic.train_lib import train_utils
  import scipy
label_desc: Use image node_id 6 for calling a global function with example usage: init_classification_bias(bias, prior_prob) and returns: jnp