Skip to content

Instantly share code, notes, and snippets.

View wasit7's full-sized avatar

Wasit Limprasert wasit7

View GitHub Profile
@wasit7
wasit7 / pykeen-ermlpe.py
Created September 5, 2023 17:20 — forked from airysen/pykeen-ermlpe.py
Example of using LM embeddings in pykeen
from pykeen.training import SLCWATrainingLoop
from pykeen.datasets import get_dataset
from pykeen.nn.representation import TextRepresentation
from pykeen.models import ERModel
from pykeen.pipeline import pipeline
from torch.optim import Adam