ontolearn.clip_architectures

Module Contents

Classes

LengthLearner_LSTM	LSTM architecture
LengthLearner_GRU	GRU architecture
LengthLearner_CNN	CNN architecture
LengthLearner_SetTransformer	SetTransformer architecture.

class ontolearn.clip_architectures.LengthLearner_LSTM(input_size, output_size, proj_dim=256, rnn_n_layers=2, drop_prob=0.2)

Bases: ontolearn.nces_modules.nn.Module

LSTM architecture

forward(x1, x2)

Forward pass through the network.

class ontolearn.clip_architectures.LengthLearner_GRU(input_size, output_size, proj_dim=256, rnn_n_layers=2, drop_prob=0.2)

Bases: ontolearn.nces_modules.nn.Module

GRU architecture

forward(x1, x2)

Forward pass through the network.

class ontolearn.clip_architectures.LengthLearner_CNN(input_size, output_size, num_examples, proj_dim=256, kernel_size: list = [[5, 7], [5, 7]], stride: list = [[3, 3], [3, 3]], drop_prob=0.2)

Bases: ontolearn.nces_modules.nn.Module

CNN architecture

forward(x1, x2)

Forward pass through the network.

class ontolearn.clip_architectures.LengthLearner_SetTransformer(input_size, output_size, proj_dim=256, num_heads=4, num_seeds=1, num_inds=32)

Bases: ontolearn.nces_modules.nn.Module

SetTransformer architecture.

forward(x1, x2)

Forward pass through the network.