e1aca5875121ef15bdb07243af56eb704d79c794,deepvariant/model_train.py,,run,#Any#Any#Any#Any#,186

Before Change

      )

      training_hooks = None
      if FLAGS.use_early_stopping:
        // Early stopping hook depends on existence of events directory.
        eval_dir = os.path.join(FLAGS.train_dir, FLAGS.early_stopping_directory)
        tf.io.gfile.makedirs(eval_dir)

        plateau_decrease = True
        if FLAGS.early_stopping_metric_direction == "increase":
          plateau_decrease = False

        early_stopping_hook = metrics_hook.EarlyStoppingHook(
            events_dir=eval_dir,
            tag=FLAGS.early_stopping_tag,
            num_plateau_steps=FLAGS.early_stopping_num_plateau_steps,
            plateau_delta=FLAGS.early_stopping_plateau_delta,
            plateau_decrease=plateau_decrease,
            every_n_steps=FLAGS.early_stopping_every_n_steps)

        training_hooks = [early_stopping_hook]

      estimator.train(
          input_fn=tf_dataset,
          max_steps=FLAGS.number_of_steps,
          hooks=training_hooks)

After Change

      )

      training_hooks = None
      if FLAGS.use_early_stopping:
        // redacted
        raise ValueError("Currently not implemented.")

      estimator.train(
          input_fn=tf_dataset,
          max_steps=FLAGS.number_of_steps,
          hooks=training_hooks)

Instances