Проблемы при использовании вспомогательных участков с желтым кирпичиком и потере легенды и заголовков - PullRequest
Новая
       42

Проблемы при использовании вспомогательных участков с желтым кирпичиком и потере легенды и заголовков

4 голосов
/ 19 апреля 2019

У меня проблемы с размещением нескольких диаграмм желтого кирпича в подпрограмме. Заголовок и легенда отображаются только для последнего графика. Я попробовал несколько способов написания кода, но не могу заставить их показать легенды и названия. Я уверен, что легко приступить к работе.

Вот кусок кода: 

f, ((ax1, ax2), (ax3, ax4)) = plt.subplots(2, 2,figsize=(14, 10))

viz = FeatureImportances(LinearRegression(), ax=ax1)
viz.fit(X_train, y_train)

viz = LearningCurve(LinearRegression(), scoring='r2',cv=10, ax=ax2)
viz.fit(X_train, y_train)

viz = ResidualsPlot(clf, ax=ax3)
viz.fit(X_train, y_train) 

viz = PredictionError(LinearRegression(), ax=ax4)
viz.fit(X_train, y_train) 
viz.score(X_test, y_test) 

viz.poof()

изображение участков

1 Ответ

4 голосов
/ 19 апреля 2019

@ chris-mangum извините, что вы боролись с этим. Помимо poof у нас есть еще один метод с именем finalize. В этом случае finalize лучше, чем poof - poof, вызывает finalize, а затем либо показывает, либо savefig, который завершает фигуру, поэтому в многоосевом графике Как и вы, вы не хотите называть пуф. 

f, ((ax1, ax2), (ax3, ax4)) = plt.subplots(2, 2,figsize=(14, 10))

viz = FeatureImportances(LinearRegression(), ax=ax1)
viz.fit(X_train, y_train)
viz.finalize()

viz = LearningCurve(LinearRegression(), scoring='r2',cv=10, ax=ax2)
viz.fit(X_train, y_train)
viz.finalize()

viz = ResidualsPlot(clf, ax=ax3)
viz.fit(X_train, y_train) 
viz.finalize()

viz = PredictionError(LinearRegression(), ax=ax4)
viz.fit(X_train, y_train) 
viz.score(X_test, y_test) 

viz.finalize()
...