Tabular transformer (part 2)

PiperOrigin-RevId: 701041195
google · Nov 28, 2024 · 4d1a42e · 4d1a42e
1 parent dd988de
commit 4d1a42e
Show file tree

Hide file tree

Showing 3 changed files with 681 additions and 480 deletions.
diff --git a/...sil_decision_forests/port/python/ydf/model/decision_forest_model/decision_forest_model.py b/...sil_decision_forests/port/python/ydf/model/decision_forest_model/decision_forest_model.py
@@ -26,7 +26,7 @@
 from ydf.model.tree import tree as tree_lib
 
 
-class DecisionForestModel(generic_model.GenericModel):
+class DecisionForestModel(generic_model.GenericCCModel):
   """A generic decision forest model for prediction and inspection."""
 
   _model: ydf.DecisionForestCCModel