How to use the keyvi.IntDictionaryCompiler function in keyvi

To help you get started, we’ve selected a few keyvi examples, based on popular ways it is used in public projects.

Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.

github KeyviDev / keyvi / python / src / py / keyvi / cli / cli.py View on Github external
def compile(args):
    params = {key: value for key, value in args.compiler_params}

    dict_type = args.dict_type
    if dict_type == 'json':
        dictionary = keyvi.JsonDictionaryCompiler(params)
    elif dict_type == 'string':
        dictionary = keyvi.StringDictionaryCompiler(params)
    elif dict_type == 'int':
        dictionary = keyvi.IntDictionaryCompiler(params)
    elif dict_type == 'completion':
        dictionary = keyvi.CompletionDictionaryCompiler(params)
    elif dict_type == 'key-only':
        dictionary = keyvi.KeyOnlyDictionaryCompiler(params)
    else:
        return 'Must never reach here'

    with open(args.input_file) as file_in:
        for line in file_in:
            line = line.rstrip('\n')
            try:
                splits = line.split('\t')
                if dict_type == 'key-only':
                    dictionary.Add(splits[0])
                elif dict_type == 'int' or dict_type == 'completion':
                    dictionary.Add(splits[0], int(splits[1]))