-
Notifications
You must be signed in to change notification settings - Fork 29
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add Forest Inference Operator (#118)
* Add backend param to export method of PipelineableInferenceOperator * Add params optional keyword argument to export method of FIL class * Add instance_group param to FIL op to allow choice of CPU or GPU * Add Forest class for running inference with forest models * Add docstrings to op runner model * Allow tensors to be returned by inference operators * Remove as_numpy from transform method of Forest operator * Add copyright to the fil and test_forest module * Add test for Ensemble with NVTabular Workflow and Forest Operator * Correct config name in forest test * Rename Forest PredictForest and update docstring. * Correct name of directory in tests * Correct backend assertion for transform op in forest ensemble test Co-authored-by: Karl Higley <kmhigley@gmail.com>
- Loading branch information
1 parent
c3f7d36
commit 253d83f
Showing
5 changed files
with
379 additions
and
14 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.