Python workflow_get_logitsの例

コード例 #1

ファイル: main.py プロジェクト: thomascherickal/pytext

def get_logits(
    context,
    model_snapshot,
    test_path,
    use_cuda,
    output_path,
    field_names,
    dump_raw_input,
    batch_size,
    ndigits_precision,
    output_columns,
    use_gzip,
    device_id,
):
    """print logits from  a trained model snapshot to output_path"""

    model_snapshot, use_cuda, _ = _get_model_snapshot(context, model_snapshot,
                                                      use_cuda, False)
    if output_columns:
        output_columns = [int(x) for x in output_columns.split(",")]

    print("\n=== Starting get_logits...")
    workflow_get_logits(
        model_snapshot,
        use_cuda,
        output_path,
        test_path,
        field_names,
        dump_raw_input,
        batch_size,
        ndigits_precision,
        output_columns,
        use_gzip,
        device_id,
    )

コード例 #2

ファイル: main.py プロジェクト: tayyabjsr13/pytext

def get_logits(context, model_snapshot, test_path, use_cuda, output_path, field_names):
    """print logits from  a trained model snapshot to output_path
    """

    model_snapshot, use_cuda, _ = _get_model_snapshot(
        context, model_snapshot, use_cuda, False
    )
    print("\n=== Starting get_logits...")
    workflow_get_logits(model_snapshot, use_cuda, output_path, test_path, field_names)