SkLearn adaboost занимает слишком много времени - PullRequest
Новая
       91

SkLearn adaboost занимает слишком много времени

0 голосов
/ 20 апреля 2020

У меня есть тренировочные данные с 401 записями, у них есть 2 функции. Есть 2 класса: 0 и 1 Я запускаю AdaBoost из SkLearn и затем пытаюсь построить границу решения. Я получаю оценку точности 94%

Но моя система зависает, когда я вызываю функцию message_function. код ниже:

Я получил код от здесь 

def plot(self):
        X = self.trainX
        y = self.trainY
        clf = AdaBoostClassifier()
        clf.fit(X, y)
        x_min, x_max = X.iloc[:, 0].min() - .5, X.iloc[:, 0].max() + .5
        y_min, y_max = X.iloc[:, 1].min() - .5, X.iloc[:, 1].max() + .5
        xx, yy = np.meshgrid(np.arange(x_min, x_max, 0.02), np.arange(y_min, y_max, 0.02))
        Z = clf.decision_function(np.c_[xx.ravel(), yy.ravel()])
        Z = Z.reshape(xx.shape)
        plt.contourf(xx, yy, Z, cmap=cm, alpha=.8)

        # Plot the training points
        plt.scatter(X_train[:, 0], X_train[:, 1], c=y_train, cmap=cm_bright, edgecolors='k')
        # Plot the testing points
        ax.scatter(X_test[:, 0], X_test[:, 1], c=y_test, cmap=cm_bright, edgecolors='k', alpha=0.6)

        plt.set_xlim(xx.min(), xx.max())
        plt.set_ylim(yy.min(), yy.max())
        plt.set_xticks(())
        plt.set_yticks(())
        plt.show()

У любого есть какое-либо представление, почему, или решение о том, как построить функцию решения.

...