neural_tangents.predict.ODEState
- class neural_tangents.predict.ODEState(fx_train=None, fx_test=None, qx_train=None, qx_test=None)[source]
ODE state dataclass holding outputs and auxiliary variables.
- __init__(fx_train=None, fx_test=None, qx_train=None, qx_test=None)
Methods
__init__([fx_train, fx_test, qx_train, qx_test])asdict(*[, dict_factory])Return the fields of a dataclass instance as a new dictionary mapping field names to field values.
astuple(*[, tuple_factory])Return the fields of a dataclass instance as a new tuple of field values.
replace(**changes)Return a new object replacing specified fields with new values.
Attributes
fx_testfx_trainqx_testqx_train