Exemple #1
0
dataloader = CombinedLoader(
    mel_data_path=prefix_path("melspec_data", common.DEFAULT_PATH),
    mel_training_path=prefix_path("autotagging_moodtheme-train.tsv",
                                  common.DEFAULT_PATH),
    mel_test_path=prefix_path("autotagging_moodtheme-test.tsv",
                              common.DEFAULT_PATH),
    mel_validate_path=prefix_path("autotagging_moodtheme-validation.tsv",
                                  common.DEFAULT_PATH),
    ess_training_path=prefix_path("accousticbrainz-train.pickle",
                                  common.DEFAULT_PATH),
    ess_test_path=prefix_path("accousticbrainz-test.pickle",
                              common.DEFAULT_PATH),
    ess_validate_path=prefix_path("accousticbrainz-validation.pickle",
                                  common.DEFAULT_PATH),
    window='random',
    num_windows=5,
)

pipeline = Pipeline([
    ("model", CRNNPlusModel(epochs=32, dataloader=dataloader)),
])

evaluator = ModelCallbackWrapper(
    FixedSplitEvaluator(**common.fixed_split_params()),
    lambda model: common.store_prediction(model, dataloader),
)

result_handlers = [
    result_handlers.print_gridsearch_results,
]
Exemple #2
0
dataloader = CombinedLoader(
    mel_data_path=prefix_path("melspec_data", common.DEFAULT_PATH),
    mel_training_path=prefix_path("autotagging_moodtheme-train.tsv",
                                  common.DEFAULT_PATH),
    mel_test_path=prefix_path("autotagging_moodtheme-test.tsv",
                              common.DEFAULT_PATH),
    mel_validate_path=prefix_path("autotagging_moodtheme-validation.tsv",
                                  common.DEFAULT_PATH),
    ess_training_path=prefix_path("accousticbrainz-train.pickle",
                                  common.DEFAULT_PATH),
    ess_test_path=prefix_path("accousticbrainz-test.pickle",
                              common.DEFAULT_PATH),
    ess_validate_path=prefix_path("accousticbrainz-validation.pickle",
                                  common.DEFAULT_PATH),
    window='random',
    num_windows=5,
)

pipeline = Pipeline([
    ("model", CRNNPlusModel(epochs=48, dataloader=dataloader, attention=True)),
])

evaluator = ModelCallbackWrapper(
    FixedSplitEvaluator(**common.fixed_split_params()),
    lambda model: common.store_prediction(model, dataloader),
)

result_handlers = [
    result_handlers.print_gridsearch_results,
]
Exemple #3
0
                                  common.DEFAULT_PATH),
    mel_test_path=prefix_path("autotagging_moodtheme-test.tsv",
                              common.DEFAULT_PATH),
    mel_validate_path=prefix_path("autotagging_moodtheme-validation.tsv",
                                  common.DEFAULT_PATH),
    ess_training_path=prefix_path("accousticbrainz-train.pickle",
                                  common.DEFAULT_PATH),
    ess_test_path=prefix_path("accousticbrainz-test.pickle",
                              common.DEFAULT_PATH),
    ess_validate_path=prefix_path("accousticbrainz-validation.pickle",
                                  common.DEFAULT_PATH),
    window='random',
    num_windows=5,
)

pipeline = Pipeline([("model", CRNNPlusModel(dataloader=dataloader))])

grid_params = common.grid_params()
grid_params['n_jobs'] = 1

evaluator = FixedSplitGridEvaluator(
    params={
        "model__epochs": [8, 16],
        "model__output_dropout": [0.3],
        "model__concat_bn": [True],
    },
    grid_params=grid_params,
)

result_handlers = [
    result_handlers.print_gridsearch_results,