File size: 248 Bytes
877add7
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
"""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()