ontolearn.nces_architectures

NCES architectures.

Classes

`LSTM`	LSTM module.
`GRU`	GRU module.
`SetTransformer`	SetTransformer module.

Module Contents

class ontolearn.nces_architectures.LSTM(knowledge_base_path, vocab, inv_vocab, max_length, input_size, proj_dim, rnn_n_layers, drop_prob)[source]

Bases: ontolearn.nces_modules.nn.Module

LSTM module.

name = 'LSTM'

max_len

proj_dim

vocab

inv_vocab

loss

lstm

bn

fc1

fc2

fc3

forward(x1, x2, target_scores=None)[source]

class ontolearn.nces_architectures.GRU(knowledge_base_path, vocab, inv_vocab, max_length, input_size, proj_dim, rnn_n_layers, drop_prob)[source]

Bases: ontolearn.nces_modules.nn.Module

GRU module.

name = 'GRU'

max_len

proj_dim

vocab

inv_vocab

loss

gru

bn

fc1

fc2

fc3

forward(x1, x2, target_scores=None)[source]

class ontolearn.nces_architectures.SetTransformer(knowledge_base_path, vocab, inv_vocab, max_length, input_size, proj_dim, num_heads, num_seeds, num_inds, ln)[source]

Bases: ontolearn.nces_modules.nn.Module

SetTransformer module.

name = 'SetTransformer'

max_len

vocab

inv_vocab

loss

enc

dec

forward(x1, x2)[source]