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_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)