Объект 'numpy .ndarray' не имеет атрибута 'tree_'

Question

Я пытаюсь построить диаграмму своего Дерева решений и получаю эту ошибку. Я искал решение, в котором говорилось, что train_test_split создает массив, и я не могу использовать эту функцию, потому что или я должен был создать массив то же имя, что и сюжет, но у меня нет

пожалуйста, помогите и расскажите, как я могу сделать этот запуск

from sklearn import tree
fn=['Pclass', 'Age', 'Sex']
cn=[0 , 1]
fig, axes = plt.subplots(nrows = 1,ncols = 1,figsize = (4,4), dpi=300)
tree.plot_tree(sTree,
               feature_names = fn, 
               class_names=cn,
               filled = True)
fig.savefig('titanic_DTree.png')

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-301-3eced642ce6e> in <module>
      5 tree.plot_tree(sTree,
      6                feature_names = fn,
----> 7                class_names=cn,
      8                );
      9 fig.savefig('titanic_DTree.png')

~\anaconda3\lib\site-packages\sklearn\tree\_export.py in plot_tree(decision_tree, max_depth, feature_names, class_names, label, filled, impurity, node_ids, proportion, rotate, rounded, precision, ax, fontsize)
    174         proportion=proportion, rotate=rotate, rounded=rounded,
    175         precision=precision, fontsize=fontsize)
--> 176     return exporter.export(decision_tree, ax=ax)
    177 
    178 

~\anaconda3\lib\site-packages\sklearn\tree\_export.py in export(self, decision_tree, ax)
    564         ax.clear()
    565         ax.set_axis_off()
--> 566         my_tree = self._make_tree(0, decision_tree.tree_,
    567                                   decision_tree.criterion)
    568         draw_tree = buchheim(my_tree)

The error I am receiving is :

AttributeError: 'numpy.ndarray' object has no attribute 'tree_'

Спасибо