powered by
Run MLP inference on embeddings
ebisg_predict_mlp(embeddings, model_path)
A matrix of dimensions (N, 6) with columns (whi, bla, his, asi, aian, oth).
Numeric matrix (N x dim) of name embeddings.
Path to the .pt checkpoint file.