plot_class_regions_for_classifier_subplot

[grjd]

Hi I'm getting this error when calling 'plot_class_regions_for_classifier_subplot'
ValueError: Number of features of the model must match the input. Model n_features is 155 and input n_features is 2

clf = GradientBoostingClassifier().fit(X_train, y_train)
fig, subaxes = plt.subplots(1,1, figsize=(6,6))
titleplot='GBDT, default settings'
plot_class_regions_for_classifier_subplot(clf, X_train, y_train, X_test, y_test,titleplot,subaxes)

%%X_train.shape (736, 155), X_test.shape (185, 155), y_train.shape (736,) y_test.shape (185,)