NVIDIA-Merlin / core

Core Utilities for NVIDIA Merlin
Apache License 2.0
19 stars 14 forks source link

Stat fit transform #332

Closed jperez999 closed 1 year ago

jperez999 commented 1 year ago

This PR adds the fitted (bool) attribute to all Stat Operators. This boolean is set to true after an operator has been fit. You can now manually specify to "refit" a fitted StatOperator in the fit method call. And transform checks to ensure all Stat Operators have been fitted before running transform on a graph.