Ejemplo n.º 1
0
 def _input_fn(params):
     return preprocessing.get_tpu_bt_input_tensors(
         games,
         games_nr,
         params['batch_size'],
         number_of_games=FLAGS.window_size,
         random_rotation=True)
Ejemplo n.º 2
0
 def _input_fn(params):
     games = bigtable_input.GameQueue(
         FLAGS.cbt_project, FLAGS.cbt_instance, FLAGS.cbt_table)
     games_nr = bigtable_input.GameQueue(
         FLAGS.cbt_project, FLAGS.cbt_instance, FLAGS.cbt_table + '-nr')
     return preprocessing.get_tpu_bt_input_tensors(
         games,
         games_nr,
         params['batch_size'],
         number_of_games=FLAGS.window_size,
         random_rotation=True)
Ejemplo n.º 3
0
 def _input_fn(params):
     return preprocessing.get_tpu_bt_input_tensors(
         params['batch_size'],
         random_rotation=True)