def main(unused_argv): tf.enable_v2_behavior() filenames = [os.path.join(FLAGS.config_path, p) for p in FLAGS.gin_config] gin.parse_config_files_and_bindings(filenames, FLAGS.gin_bindings) seed = FLAGS.seed if FLAGS.seed is not None else int(time.time()) tf.random.set_seed(seed) training.TrainingLoop(workdir=FLAGS.base_dir).run()
def test_train(self):
factorizer = nmf.NMF(low_rank=6, num_iterations=20)
loop = training.TrainingLoop(None,
data_loader=self.synthetizer,
factorizer=factorizer)
loop.run(steps=2)
self.assertEqual(factorizer.u.shape, (self.num_features, self.rank))
self.assertEqual(factorizer.v.shape, (self.rank, self.num_individuals))