← all blocks
model-head · v1.0.0

Set Transformer (TF.js)

Permutation-invariant set encoder. Element-wise embed + self-attention + mean-pool. Toy task: sum-of-set binary classification on N=10 reals.

Contract

Input:
Frame
Output:
ModelArtifact
Determinism:
seeded
Side effects:
trains-artifact
Leakage window:
0s

Params (live form)

JSON
{}
Try in builder