ошибка при отображении модели дерева в графвизе - PullRequest
0 голосов
/ 07 марта 2019

Я пытаюсь отобразить модель случайного леса в своей записной книжке на компьютере. И получить следующее сообщение об ошибке. Любая помощь с этим.

m = RandomForestRegressor(n_estimators=1, max_depth=3, bootstrap=False, n_jobs=-1)
m.fit(X_train, y_train)

t=m.estimators_[0]
size=10
ratio=0.6
s=export_graphviz(t, out_file=None, feature_names=df_trn.columns, filled=True,
                      special_characters=True, rotate=True, precision=3)
IPython.display.display(graphviz.Source(re.sub('Tree {',f'Tree {{ size={size}; ratio={ratio}', s)))

Сообщение об ошибке

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-38-12fbbbfe5f78> in <module>
----> 1 IPython.display.display(graphviz.Source(re.sub('Tree {',f'Tree {{ size={size}; ratio={ratio}', s)))

~/anaconda/envs/fastai/lib/python3.6/site-packages/graphviz/files.py in __init__(self, source, filename, directory, format, engine, encoding)
    273     def __init__(self, source, filename=None, directory=None,
    274                  format=None, engine=None, encoding=File._encoding):
--> 275         super(Source, self).__init__(filename, directory, format, engine, encoding)
    276         self.source = source  #: The verbatim DOT source code string.
    277 

TypeError: super(type, obj): obj must be an instance or subtype of type
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...