mirror of
https://github.com/explosion/spaCy.git
synced 2024-11-11 12:18:04 +03:00
44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
# coding: utf8
|
|
from __future__ import unicode_literals
|
|
|
|
import plac
|
|
from pathlib import Path
|
|
|
|
from .converters import conllu2json, iob2json
|
|
from ..util import prints
|
|
|
|
# Converters are matched by file extension. To add a converter, add a new entry
|
|
# to this dict with the file extension mapped to the converter function imported
|
|
# from /converters.
|
|
|
|
CONVERTERS = {
|
|
'.conllu': conllu2json,
|
|
'.conll': conllu2json,
|
|
'.iob': iob2json
|
|
}
|
|
|
|
|
|
@plac.annotations(
|
|
input_file=("input file", "positional", None, str),
|
|
output_dir=("output directory for converted file", "positional", None, str),
|
|
n_sents=("Number of sentences per doc", "option", "n", int),
|
|
morphology=("Enable appending morphology to tags", "flag", "m", bool)
|
|
)
|
|
def convert(cmd, input_file, output_dir, n_sents=1, morphology=False):
|
|
"""
|
|
Convert files into JSON format for use with train command and other
|
|
experiment management functions.
|
|
"""
|
|
input_path = Path(input_file)
|
|
output_path = Path(output_dir)
|
|
if not input_path.exists():
|
|
prints(input_path, title="Input file not found", exits=1)
|
|
if not output_path.exists():
|
|
prints(output_path, title="Output directory not found", exits=1)
|
|
file_ext = input_path.suffix
|
|
if not file_ext in CONVERTERS:
|
|
prints("Can't find converter for %s" % input_path.parts[-1],
|
|
title="Unknown format", exits=1)
|
|
CONVERTERS[file_ext](input_path, output_path,
|
|
n_sents=n_sents, use_morphology=morphology)
|