search

MichSchli / RelationPrediction

Implementation of R-GCNs for Relational Link Prediction
MIT License
435 stars 103 forks source link

Dimension of relation_shape #5

Open axelreichwein opened 6 years ago

axelreichwein commented 6 years ago

The dimension of relation_shape is set based on EntityCount in https://github.com/MichSchli/RelationPrediction/blob/master/code/common/model_builder.py#L83

Shouldn't it rather be based on RelationCount?