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