Esempio n. 1
0
    def repeatTheLearningProcess(self, bestGridSearched, set):
        _, bestParams = self.getBestGridSearchedModel(bestGridSearched, set)

        if bestGridSearched.learnerType == 'KNN':
            bestGridSearched = KNN.KNNLearner(**bestParams,
                                              datasetNo=set.datasetNo)
        elif bestGridSearched.learnerType == 'DT':
            bestGridSearched = DT.DTLearner(**bestParams,
                                            datasetNo=set.datasetNo)
        elif bestGridSearched.learnerType == 'SVM':
            bestGridSearched = SVM.SVMLearner(**bestParams,
                                              datasetNo=set.datasetNo)
        elif bestGridSearched.learnerType == 'Boosting':
            bestGridSearched = Boosting.BoostingLearner(
                **bestParams, datasetNo=set.datasetNo)
        elif bestGridSearched.learnerType == 'ANN':
            bestGridSearched = ANN.ANNLearner(**bestParams,
                                              datasetNo=set.datasetNo)

        self.getLearningCurve(bestGridSearched, set)
        self.getComplexityCurve(bestGridSearched, set)

        return bestGridSearched