Я тоже провел некоторое время с этим, поскольку plot_decision_regions
тогда жаловался ValueError: Column(s) [2] need to be accounted for in either feature_index or filler_feature_values
, и есть еще один параметр, необходимый, чтобы избежать этого.
Итак, скажем, у вас есть 4 объекта, и они не имеют названия:
X_train_std.shape[1] = 4
Мы можем ссылаться на каждую функцию по их индексу 0, 1, 2, 3. Вы можете построить только 2функции, скажем, вы хотите 0
и 2
.
Вам нужно будет указать один дополнительный параметр (к тому, который указан в ответе @ sos.cott), feature_index и заполнить остальные заполнителями:
value=1.5
width=0.75
fig = plot_decision_regions(X_train.values, y_train.values, clf=clf,
feature_index=[0,2], #these one will be plotted
filler_feature_values={1: value, 3:value}, #these will be ignored
filler_feature_ranges={1: width, 3: width})