def vary_shuffle(argTest=True):
    decodeConfig = decode_config.DecodeConfig()
    decodeConfig.decode_style = 'valence'

    decodeConfig.repeat = 10
    decodeConfig.neurons = 40
    decodeConfig.average_time = True

    hp_ranges = OrderedDict()
    hp_ranges['shuffle'] = [False, True]
    if argTest:
        decodeConfig.repeat = 5
    return decodeConfig, hp_ranges
def vary_neuron(argTest=True, neurons=50, style=('valence')):
    decodeConfig = decode_config.DecodeConfig()
    decodeConfig.shuffle = False
    decodeConfig.repeat = 100
    decodeConfig.average_time = True
    hp_ranges = OrderedDict()
    hp_ranges['neurons'] = np.arange(5, neurons, 5)
    hp_ranges['decode_style'] = style
    hp_ranges['shuffle'] = [False, True]
    if argTest:
        decodeConfig.repeat = 50
        hp_ranges['neurons'] = np.arange(10, neurons, 10)
    return decodeConfig, hp_ranges
def vary_decode_style(argTest=True,
                      style=('identity', 'csp_identity', 'csm_identity',
                             'valence')):
    decodeConfig = decode_config.DecodeConfig()
    decodeConfig.repeat = 100
    decodeConfig.neurons = 40
    decodeConfig.average_time = True

    hp_ranges = OrderedDict()
    hp_ranges['shuffle'] = [False, True]
    hp_ranges['decode_style'] = style
    if argTest:
        decodeConfig.repeat = 50
    return decodeConfig, hp_ranges
def test_across_days(argTest=True,
                     neurons=40,
                     style=('identity'),
                     no_end_time=True):
    decodeConfig = decode_config.DecodeConfig()
    decodeConfig.repeat = 100
    decodeConfig.neurons = neurons
    decodeConfig.average_time = True
    decodeConfig.no_end_time = no_end_time
    hp_ranges = OrderedDict()
    hp_ranges['decode_style'] = style
    hp_ranges['shuffle'] = [True, False]

    if argTest:
        decodeConfig.repeat = 50
    return decodeConfig, hp_ranges
def test_split(argTest=True,
               neurons=40,
               style=('csp_identity'),
               no_end_time=False,
               start_day=[],
               end_day=[]):
    decodeConfig = decode_config.DecodeConfig()
    decodeConfig.repeat = 100
    decodeConfig.neurons = neurons
    decodeConfig.average_time = True
    decodeConfig.no_end_time = no_end_time
    decodeConfig.start_day = start_day
    decodeConfig.end_day = end_day
    decodeConfig.shuffle = False
    hp_ranges = OrderedDict()
    hp_ranges['decode_style'] = style
    hp_ranges['split_style'] = ['onset', 'time', 'magnitude']

    if argTest:
        decodeConfig.repeat = 50
    return decodeConfig, hp_ranges
def test_fp_fn(argTest=True,
               neurons=40,
               style=('valence'),
               no_end_time=False,
               start_day=[],
               end_day=[],
               res_counting=None):
    decodeConfig = decode_config.DecodeConfig()
    decodeConfig.repeat = 100
    decodeConfig.neurons = neurons
    decodeConfig.average_time = True
    decodeConfig.no_end_time = no_end_time
    decodeConfig.start_day = start_day
    decodeConfig.end_day = end_day
    decodeConfig.res_counting = res_counting
    hp_ranges = OrderedDict()
    hp_ranges['decode_style'] = style
    hp_ranges['shuffle'] = [False, True]

    if argTest:
        decodeConfig.repeat = 50
    return decodeConfig, hp_ranges