CompGCN

class CompGCN(*, triples_factory, embedding_dim=64, encoder_kwargs=None, interaction=None, interaction_kwargs=None, **kwargs)[source]

Bases: ERModel[FloatTensor, RelationRepresentation, FloatTensor]

An implementation of CompGCN from [vashishth2020].

This model uses graph convolutions, and composition functions.

Initialize the model.

Parameters:

Attributes Summary

hpo_default

The default strategy for optimizing the model's hyper-parameters

Attributes Documentation

hpo_default: ClassVar[Mapping[str, Any]] = {'embedding_dim': {'high': 512, 'low': 32, 'q': 32, 'type': <class 'int'>}}

The default strategy for optimizing the model’s hyper-parameters