adithya9903's picture
Deploy PolyGuard HF training Space
fd0c71a verified
"""Regimen embedding helper."""
from __future__ import annotations
import numpy as np
from app.models.graph.hetero_encoder import encode_regimen
def regimen_embedding(drugs: list[str]) -> list[float]:
return encode_regimen(drugs).tolist()