Learn R Programming

wru (version 3.1.0)

ebisg_predict_mlp: Run MLP inference on embeddings

Description

Run MLP inference on embeddings

Usage

ebisg_predict_mlp(embeddings, model_path)

Value

A matrix of dimensions (N, 6) with columns (whi, bla, his, asi, aian, oth).

Arguments

embeddings

Numeric matrix (N x dim) of name embeddings.

model_path

Path to the .pt checkpoint file.