def extract_type_from_nnlm(data):
    """
    Extract the version for the NNLM collection of Language Models
    https://tfhub.dev/google/collections/nnlm/1
    """
    return data['lm'].str.extract(r'nnlm-(e.-.*)-w.*')