_task should be task (now?) in tensorflow trees

parrt / dtreeviz

A python library for decision tree visualization and model interpretation.

MIT License

2.89k stars 333 forks source link

_task should be task (now?) in tensorflow trees #256

Closed parrt closed 1 year ago

parrt commented 1 year ago

[/usr/local/lib/python3.8/dist-packages/dtreeviz/models/tensorflow_decision_tree.py](https://localhost:8080/#) in is_classifier(self)
     56 
     57     def is_classifier(self) -> bool:
---> 58         return self.model._task == Task.CLASSIFICATION
     59 
     60     def get_class_weights(self):

AttributeError: 'RandomForestModel' object has no attribute '_task'