INNER CODE UNIT · Python

_parse_values

taishi-i/nagisa · nagisa/dynet_loader.py:26

def _parse_values(data):
    text = data.decode('latin-1')
    try:
        return np.fromstring(text, dtype=np.float64, sep=' ')
    except (AttributeError, ValueError):
        # np.fromstring's text mode may go away in future numpy releases.
        return np.array(text.split(), dtype=np.float64)


def load_dynet_text_model(path):
    """Parse a DyNet text-format model file.

    return:
        - OrderedDict mapping parameter name to (kind, array) where kind is
          "Parameter" or "LookupParameter". Plain parameters keep DyNet's
          dim layout (e.g. (6, 100)); lookup parameters are returned as
          (num_entries, dim), the same layout as LookupParameters.as_array().
    """

View source record →