code: def compute_mask_prob_from_yao_schedule(
      i, n, pmin=0.1, pmax=0.9, alpha=0.7
  ):
      wat = (pmax - pmin) * i / n
      return tf.maximum(pmin, pmax - wat / alpha)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import time
  from magenta.models.coconet import lib_graph
  from magenta.models.coconet import lib_hparams
  import numpy
  import tensorflow.compat.v1
label_desc: Use image node_id 4 for calling a global function with example usage: compute_mask_prob_from_yao_schedule(i, n, pmin, pmax, alpha) and returns: tf