Exemplo n.º 1
0
def get_num_features(data_type, corpus_file, side):
    """
    Args:
        data_type (str): type of the source input.
            Options are [text|img|audio].
        corpus_file (str): file path to get the features.

    Returns:
        number of features on `side`.
    """
    # if data_type == 'concat':
    return TextDataset.get_num_features(corpus_file, side)
Exemplo n.º 2
0
def get_num_features(data_type, corpus_file, side):
    """
    Args:
        data_type (str): type of the source input.
            Options are [text|img|audio].
        corpus_file (str): file path to get the features.
        side (str): for source or for target.

    Returns:
        number of features on `side`.
    """
    assert side in ["src", "tgt"]

    return TextDataset.get_num_features(corpus_file, side)
Exemplo n.º 3
0
def get_num_features(data_type, corpus_file, side):
    """
    Args:
        data_type (str): type of the source input.
            Options are [text|img|audio].
        corpus_file (str): file path to get the features.
        side (str): for source or for target.

    Returns:
        number of features on `side`.
    """
    assert side in ["src", "tgt", "ans"]

    if data_type == 'text':
        return TextDataset.get_num_features(corpus_file, side)
    else:
        raise ValueError("Data type not implemented")