tensorflow / decision-forests

A collection of state-of-the-art algorithms for the training, serving and interpretation of Decision Forest models in Keras.
Apache License 2.0
663 stars 110 forks source link

Fixed bug in dataframe to tf dataset conversion #53

Closed rishiraj closed 2 years ago

rishiraj commented 3 years ago

The script was converting the train pandas dataframe (train_ds_pd) into test tensorflow datasets (test_ds). Fixed it, now test pandas dataframe is converted to test tensorflow dataset.

review-notebook-app[bot] commented 3 years ago

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

janpfeifer commented 3 years ago

Many thanks for the fix @rishiraj ! We'll include this on the next release -- it is somewhat delayed though due to the change in ABI issues with TF2.6. We'll keep you posted.

rishiraj commented 2 years ago

Any update to this @janpfeifer ?

achoum commented 2 years ago

Hi @rishiraj,

Thanks for the fix, for the pr, and for pinging us back! You change was integrated.

M.