示例#1
0
    def gaussian_process(self, label, result_list):
        clf = GaussianProcessClassifier()

        return execute_predict_proba(clf, self.train_test_split, label, result_list, self.image_creator)
示例#2
0
    def mlp(self, label, result_list):
        clf = MLPClassifier()

        return execute_predict_proba(clf, self.train_test_split, label, result_list, self.image_creator)
示例#3
0
    def xgboost(self, label, result_list):
        clf = xgb.XGBClassifier(max_depth=10)

        return execute_predict_proba(clf, self.train_test_split, label, result_list, self.image_creator)
示例#4
0
    def gnb(self, label, result_list):
        clf = GaussianNB()

        return execute_predict_proba(clf, self.train_test_split, label, result_list, self.image_creator)
示例#5
0
    def random_forest(self, label, result_list):
        clf = ensemble.RandomForestClassifier()

        return execute_predict_proba(clf, self.train_test_split, label, result_list, self.image_creator)
示例#6
0
    def decision_tree(self, label, result_list):
        clf = tree.DecisionTreeClassifier()

        return execute_predict_proba(clf, self.train_test_split, label, result_list, self.image_creator)
示例#7
0
    def knn(self, label, result_list):
        clf = neighbors.KNeighborsClassifier(n_neighbors=3, weights='distance')

        return execute_predict_proba(clf, self.train_test_split, label, result_list, self.image_creator)