예제 #1
0
def expt_args(parser):
    parser = base_expt_args(parser)
    expt_parser = parser.add_argument_group("Elderly Experiment arguments")
    expt_parser.add_argument('--n-parallel', default=1, type=int)
    expt_parser.add_argument('--mlflow', action='store_true')
    expt_parser.add_argument('--n-classes', default=2, type=int)

    return parser
예제 #2
0
def expt_args(parser):
    parser = base_expt_args(parser)
    expt_parser = parser.add_argument_group(
        "ASD ultrasound Experiment arguments")
    expt_parser.add_argument('--n-parallel', default=1, type=int)
    expt_parser.add_argument('--mlflow', action='store_true')
    expt_parser.add_argument('--notify-slack', action='store_true')

    return parser
예제 #3
0
def mask_expt_args(parser):
    parser = base_expt_args(parser)
    expt_parser = parser.add_argument_group("Mask Experiment arguments")
    expt_parser.add_argument('--target', default='normal', choices=AVAILABLE_TARGET)
    expt_parser.add_argument('--n-waves', help='Number of wave files to make one instance', type=int, default=1)
    expt_parser.add_argument('--shuffle-order', action='store_true', default=False,
                             help='Shuffle wave orders on multiple waves or not')
    expt_parser.add_argument('--n-parallel', default=1, type=int)
    expt_parser.add_argument('--only-test', action='store_true')

    return parser
예제 #4
0
def elderly_expt_args(parser):
    parser = base_expt_args(parser)
    expt_parser = parser.add_argument_group("Elderly Experiment arguments")
    expt_parser.add_argument('--target',
                             help='Valence or arousal',
                             default='valence,arousal',
                             type=type_float_list)
    expt_parser.add_argument('--parallel',
                             help='Parallel hyperparameter execution',
                             action='store_true')
    expt_parser.add_argument('--n-waves',
                             help='Number of wave files to make one instance',
                             type=int,
                             default=1)
    expt_parser.add_argument(
        '--shuffle-order',
        action='store_true',
        default=False,
        help='Shuffle wave orders on multiple waves or not')

    return parser
예제 #5
0
def elderly_expt_args(parser):
    parser = base_expt_args(parser)
    expt_parser = parser.add_argument_group("Elderly Experiment arguments")
    expt_parser.add_argument('--target',
                             help='Valence or arousal',
                             default='valence',
                             choices=AVAILABLE_TARGET)
    expt_parser.add_argument('--n-waves',
                             help='Number of wave files to make one instance',
                             type=int,
                             default=1)
    expt_parser.add_argument(
        '--shuffle-order',
        action='store_true',
        default=False,
        help='Shuffle wave orders on multiple waves or not')
    expt_parser.add_argument('--n-parallel', default=1, type=int)
    expt_parser.add_argument('--smoothing',
                             default='majority',
                             choices=['majority'])

    return parser