Skip to content
Merged
Prev Previous commit
Next Next commit
Removed tokenizer from tabular data example
  • Loading branch information
gustavocidornelas committed Jul 25, 2022
commit 4daa5f03e81fedbe9b910518b3cff08b08613b2f
2 changes: 1 addition & 1 deletion unboxapi/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -327,7 +327,7 @@ def add_model(
The ``model`` arg must be the actual trained model object, and the ``input_features`` arg must be a 2D numpy array
containing a batch of features that will be passed to the model as inputs.

You can optionally include other kwargs in the function, including tokenizers, variables, encoders etc.
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

curious why you removed tokenizers?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Oh, just because this is inside the example for tabular classification. I kept it for the example for NLP.

I'm trying to think now of uses of a tokenizer on a tabular task type... I think it would be a bit unusual, but I guess it can exist. I'll add it back just in case

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Oh no, you're right. Keep it removed for tabular :)

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Ok!

You can optionally include other kwargs in the function, including variables, encoders etc.
You simply pass those kwargs to the ``project.add_model`` function call when you upload the model.

Here's an example of the ``predict_proba`` function in action:
Expand Down