Участок от Мглеарн

Question

Как я могу сохранить сюжет из mglearn? Я пробовал этот код, но он не работает.

import numpy as np
import matplotlib.pyplot as plt
import mglearn
from fpdf import FPDF


f = mglearn.plots.plot_knn_classification(n_neighbors=3)
f.savefig("n_neighbors.pdf", bbox_inches='tight')

Ошибка была AttributeError: у объекта 'NoneType' нет атрибута 'savefig'. Спасибо

Answer 1

from sklearn.neighbors import KNeighborsClassifier
import matplotlib.pyplot as plt
import mglearn.plots

import numpy as np
import mglearn
from fpdf import FPDF

X, y = mglearn.datasets.make_forge()


mglearn.plots.plot_knn_classification(n_neighbors=3)
plt.savefig("n_neighbors.pdf", bbox_inches='tight')

Этот код работает.

Answer 2

Поскольку пакет mglearn использует механику построения matplotlib.pyplot, вы можете использовать их спасительную механику, например так.

mglearn.plots.plot_knn_classification(n_neighbors=3)
plt.savefig("n_neighbors.pdf", bbox_inches='tight')