Module pywander.models

Functions

def get_models_path(*args, app_name='test')

Expand source code

def get_models_path(*args, app_name='test'):
    """
    获取模型文件路径
    """
    if not args:
        raise Exception('please input the model filename.')

    path = normalized_path(os.path.join('~', 'Pywander', app_name, 'models', *args))

    return path

获取模型文件路径

def load_model(*args, app_name='test')

Expand source code

def load_model(*args, app_name='test'):
    """

    """
    model_path = get_models_path(*args, app_name=app_name)

    with open(model_path, 'rb') as f:
        # 将自动检测所使用的协议版本，因此我们
        # 不需要指定它。
        model = dill.load(f)

    logger.info(f'load model from: {model_path}')
    return model

def save_model(model, *args, app_name='test')

Expand source code

def save_model(model, *args, app_name='test'):
    """

    """
    model_path = get_models_path(*args, app_name=app_name)

    with open(model_path, 'wb') as f:
        dill.dump(model, f, recurse=True)

    logger.info(f'model has saved to: {model_path}')