:py:mod:`ontolearn.nces_architectures` ====================================== .. py:module:: ontolearn.nces_architectures .. autoapi-nested-parse:: NCES architectures. Module Contents --------------- Classes ~~~~~~~ .. autoapisummary:: ontolearn.nces_architectures.LSTM ontolearn.nces_architectures.GRU ontolearn.nces_architectures.SetTransformer .. py:class:: LSTM(knowledge_base_path, vocab, inv_vocab, max_length, input_size, proj_dim, rnn_n_layers, drop_prob) Bases: :py:obj:`ontolearn.nces_modules.nn.Module` LSTM module. .. py:method:: forward(x1, x2, target_scores=None) .. py:class:: GRU(knowledge_base_path, vocab, inv_vocab, max_length, input_size, proj_dim, rnn_n_layers, drop_prob) Bases: :py:obj:`ontolearn.nces_modules.nn.Module` GRU module. .. py:method:: forward(x1, x2, target_scores=None) .. py:class:: SetTransformer(knowledge_base_path, vocab, inv_vocab, max_length, input_size, proj_dim, num_heads, num_seeds, num_inds, ln) Bases: :py:obj:`ontolearn.nces_modules.nn.Module` SetTransformer module. .. py:method:: forward(x1, x2)