def get_glass_options(session_tracker): cases = DataSets.Glass() lrate = 0.001 cman = CustomCaseManager(cases, None, cfrac=1, vfrac=0.1, tfrac=0.1) params = { 'net_dims': calc_net_dims(cases[0], [80, 80]), 'h_activation_function': tf.nn.relu, 'o_activation_function': tf.nn.softmax, 'cost_function': tf.losses.softmax_cross_entropy, 'learning_rate': lrate, 'weight_range': [-.5, .5], 'optimizer': tf.train.AdamOptimizer(lrate, 0.9, 0.999), 'case_manager': cman, 'minibatch_size': 75, 'steps': 1000, 'vint': 10, 'session_tracker': session_tracker, # MAP options 'map_case_count': 10, 'map_case_func': cman.get_testing_cases } # Options used throughout the program return Options(**params)
def get_bit_counter_options(session_tracker): cases = DataSets.Bit_Counter(500, 15) lrate = 0.1 cman = CustomCaseManager(cases, None, cfrac=1, vfrac=0.1, tfrac=0.1) params = { 'net_dims': calc_net_dims(cases[0], [80, 80]), 'h_activation_function': tf.nn.relu, 'o_activation_function': tf.nn.softmax, 'cost_function': tf.losses.mean_squared_error, 'learning_rate': lrate, 'weight_range': [-.2, .2], 'optimizer': tf.train.AdagradOptimizer(lrate, 0.001), 'case_manager': cman, 'minibatch_size': 100, 'steps': 10000, 'vint': 100, 'session_tracker': session_tracker, # MAP options 'map_case_count': 10, 'map_case_func': cman.get_testing_cases } # Options used throughout the program return Options(**params)
def get_custom_options(net_dims, h_activation_function, o_activation_function, cost_function, learning_rate, weight_range, optimizer, minibatch_size, steps, vint, session_tracker, case_manager, map_case_count, map_case_func): params = { 'net_dims': calc_net_dims(case_manager.get_training_cases()[0], net_dims), 'h_activation_function': h_activation_function, 'o_activation_function': o_activation_function, 'cost_function': cost_function, 'learning_rate': learning_rate, 'weight_range': weight_range, 'optimizer': optimizer, 'case_manager': case_manager, 'minibatch_size': minibatch_size, 'steps': steps, 'vint': vint, 'session_tracker': session_tracker, # MAP options 'map_case_count': map_case_count, 'map_case_func': map_case_func } return Options(**params)