Exemplo n.º 1
0
    def __init__(
        self,
        trainingData=None,
        validationData=None,
        classifierType="perceptron",
        agentToClone=None,
        numTraining=3,
    ):
        from dataClassifier import (enhancedFeatureExtractorPacman,
                                    runClassifier)

        legalLabels = ["Stop", "West", "East", "North", "South"]
        if classifierType == "perceptron":
            classifier = perceptron_pacman.PerceptronClassifierPacman(
                legalLabels, numTraining
            )
        self.classifier = classifier
        self.featureFunction = enhancedFeatureExtractorPacman
        args = {
            "featureFunction": self.featureFunction,
            "classifier": self.classifier,
            "printImage": None,
            "trainingData": trainingData,
            "validationData": validationData,
            "agentToClone": agentToClone,
        }
        options = DummyOptions()
        options.classifier = classifierType
        runClassifier(args, options)
Exemplo n.º 2
0
 def __init__(self, trainingData=None, validationData=None, classifierType="perceptron", agentToClone=None, numTraining=3):
     self.classifier = classifier
     self.featureFunction = enhancedFeatureExtractorPacman
     args = {'featureFunction': self.featureFunction,
         'classifier':self.classifier,
             'printImage':None,
             'trainingData':trainingData,
             'validationData':validationData,
             'agentToClone': agentToClone,
     }
     from dataClassifier import runClassifier, enhancedFeatureExtractorPacman
     legalLabels = ['Stop', 'West', 'East', 'North', 'South']
     if(classifierType == "perceptron"):
         classifier = perceptron_pacman.PerceptronClassifierPacman(legalLabels,numTraining)
     options = DummyOptions()
     options.classifier = classifierType
     runClassifier(args, options)
 def __init__(self, trainingData, validationData, classifierType, agentToClone, numTraining=3):
     from dataClassifier import runClassifier, enhancedFeatureExtractorPacman
     legalLabels = ['Stop', 'West', 'East', 'North', 'South']
     if(classifierType == "perceptron"):
         classifier = perceptron_pacman.PerceptronClassifierPacman(legalLabels,numTraining)
     self.classifier = classifier
     self.featureFunction = enhancedFeatureExtractorPacman
     args = {'featureFunction': self.featureFunction,
             'classifier':self.classifier,
             'printImage':None,
             'trainingData':trainingData,
             'validationData':validationData,
             'agentToClone': agentToClone,
     }
     options = DummyOptions()
     options.classifier = classifierType
     runClassifier(args, options)
    def __init__(self,
                 trainingData=None,
                 validationData=None,
                 classifierType="perceptron",
                 agentToClone=None,
                 numTraining=3):
        from dataClassifier import runClassifier, enhancedFeatureExtractorPacman
        legalLabels = ['Stop', 'West', 'East', 'North', 'South']

        didLoadFromFile = False
        self.classifierType = classifierType
        if (classifierType == "perceptron"):
            try:
                classifier = pickle.load(open("perceptron.pkl", "rb"))
                print("loaded perceptron from file")
                didLoadFromFile = True
            except:
                # here's the actual perceptron part
                print("perceptron.pkl not found")
                classifier = perceptron_pacman.PerceptronClassifierPacman(
                    legalLabels, numTraining)

        self.classifier = classifier
        #looks like enhanced Feature extractor is in dataClassifier
        self.featureFunction = enhancedFeatureExtractorPacman
        args = {
            'featureFunction': self.featureFunction,
            'classifier': self.classifier,
            'printImage': None,
            'trainingData': trainingData,
            'validationData': validationData,
            'agentToClone': agentToClone,
        }
        options = DummyOptions()
        options.classifier = classifierType
        runClassifier(args, options, didLoadFromFile=didLoadFromFile)

        if not didLoadFromFile:
            pickle.dump(classifier, open("perceptron.pkl", "wb"))
            print("saving perceptron to file")