ytm._core.initializer.TAInit

class ytm._core.initializer.TAInit

Bases: ABC

Base class for TA initializers, passed as ta_init to a model.

Subclasses implement __call__, which gets the model’s RNG, its config and the initial clause weights of shape (n_classes, n_clauses), and returns the initial TA states of shape (total_clauses, n_literals).