from soweego.linker import train
LOGGER = logging.getLogger(__name__)
# Let the user pass extra kwargs to the classifier
# This is for development purposes only, and is not explicitly documented
@click.command(context_settings={
'ignore_unknown_options': True,
'allow_extra_args': True
})
@click.argument('classifier', type=click.Choice(constants.CLASSIFIERS))
@click.argument('catalog',
type=click.Choice(target_database.supported_targets()))
@click.argument('entity',
type=click.Choice(target_database.supported_entities()))
@click.option('-k',
'--k-folds',
default=5,
help="Number of folds, default: 5.")
@click.option(
'-s',
'--single',
is_flag=True,
help='Compute a single evaluation over all k folds, instead of k '
'evaluations.',
)
@click.option(
'-n',
'--nested',
is_flag=True,
f'{keys.MUSICBRAINZ}_{keys.MUSICIAN}': MUSICBRAINZ_PERSON_URL,
f'{keys.MUSICBRAINZ}_{keys.BAND}': MUSICBRAINZ_PERSON_URL,
f'{keys.MUSICBRAINZ}_{keys.MUSICAL_WORK}': MUSICBRAINZ_WORK_URL,
f'{keys.IMDB}_{keys.ACTOR}': IMDB_PERSON_URL,
f'{keys.IMDB}_{keys.DIRECTOR}': IMDB_PERSON_URL,
f'{keys.IMDB}_{keys.MUSICIAN}': IMDB_PERSON_URL,
f'{keys.IMDB}_{keys.PRODUCER}': IMDB_PERSON_URL,
f'{keys.IMDB}_{keys.WRITER}': IMDB_PERSON_URL,
f'{keys.IMDB}_{keys.AUDIOVISUAL_WORK}': IMDB_WORK_URL,
keys.TWITTER: TWITTER_URL,
}
@click.command()
@click.argument('catalog', type=click.Choice(SUPPORTED_TARGETS))
@click.argument('entity', type=click.Choice(target_database.supported_entities()))
@click.argument('confidence_range', type=(float, float))
@click.argument('matches', type=click.Path(exists=True, dir_okay=False))
def cli(catalog, entity, confidence_range, matches):
"""Upload matches to the Mix'n'match tool.
CONFIDENCE_RANGE must be a pair of floats
that indicate the minimum and maximum confidence scores.
MATCHES must be a CSV file path.
Format: QID, catalog_identifier, confidence_score
The CSV file can be compressed.
Example:
