• StreamE
• • 安装
  • 安装CUDA和cudann：
  • 训练和测试

StreamE

StreamE是一个轻量级框架，用于在流式场景中高效生成时序知识图谱（Temporal Knowledge Graph, TKG）表示。为了减小参数大小，在StreamE中，实体表示与模型训练解耦，以充当存储实体历史信息的内存模块。为了实现高效的更新和生成，在StreamE中将生成表示的过程解耦为两个函数。一个更新函数用于根据新到达的知识增量更新实体表示，另一个读取函数用于预测实体表示的未来语义。
本项目出自：电子科技大学 邵杰老师团队。

安装

安装以下包：

pip install torch
pip install numpy

安装CUDA和cudann：

pip install cutorch
pip install cunn
pip install cudnn

训练和测试

python Main.py -dataset ICEWS14