def test_end_to_end_while_loop(self):
# We cannot set use_while_loop = True in the flagsaver constructor, because
# if the flagsaver sets it to True before setting use_estimator to False,
# the flag validator will throw an error.
flags.FLAGS.use_while_loop = True
ncf_main.main(None)
flags.FLAGS.ml_perf = True
ncf_main.main(None)
def test_end_to_end_while_loop(self): # We cannot set use_while_loop = True in the flagsaver constructor, because # if the flagsaver sets it to True before setting use_estimator to False, # the flag validator will throw an error. flags.FLAGS.use_while_loop = True ncf_main.main(None) flags.FLAGS.ml_perf = True ncf_main.main(None)
def test_end_to_end_mlperf(self):
ncf_main.main(None)
def test_end_to_end(self):
ncf_main.main(None)
def test_end_to_end_mlperf(self): ncf_main.main(None)
def test_end_to_end(self): ncf_main.main(None)
def test_end_to_end_no_estimator(self):
ncf_main.main(None)
flags.FLAGS.ml_perf = True
ncf_main.main(None)
def test_end_to_end_no_estimator(self): ncf_main.main(None) flags.FLAGS.ml_perf = True ncf_main.main(None)
def test_end_to_end_while_loop(self):
ncf_main.main(None)
flags.FLAGS.ml_perf = True
ncf_main.main(None)