使用场景
使用预训练检查点ultra_4g.pth在WN18RR数据集上进行零样本推理:
python run.py -c config/transductive/inference.yaml --dataset WN18RR --epochs 0 --ckpt ckpts/ultra_4g.pth
在自定义知识图谱上微调预训练检查点ultra_4g.pth:
python run.py -c config/custom_graph.yaml --dataset MyGraph --epochs 20 --ckpt ckpts/ultra_4g.pth
产品特色
可以使用提供的预训练检查点对任何图谱(包括自定义图谱)进行零样本推理和微调
支持多GPU训练和推理
可以在自定义的图谱混合上预训练ULTRA
可以顺序评估多个数据集
可以使用预训练检查点对自定义知识图谱进行推理和微调
