ontolearn.nces_architectures

NCES architectures.

Module Contents

Classes

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

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

Bases: ontolearn.nces_modules.nn.Module

LSTM module.

forward(x1, x2, target_scores=None)

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

Bases: ontolearn.nces_modules.nn.Module

GRU module.

forward(x1, x2, target_scores=None)

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

Bases: ontolearn.nces_modules.nn.Module

SetTransformer module.

forward(x1, x2)