lace.engine.Engine.row_assignments

Engine.row_assignments(state_ix: int)

Return the assignment of rows to categories for each view.

Parameters:

state_ix (int) – The state index for which to pull the column assignment

Returns:

asgnasgn[j][i] is the index of the category to which row i is assigned under view j.

Return type:

List[List[int]]

Examples

Get the assignment category index of row 11 in view 1 of state 0

>>> from lace.examples import Animals
>>> animals = Animals()
>>> animals.row_assignments(0)[1][11]
3