dac.py 文件源码

def __init__(self, model=None, debug=False, features=False,
        candidates=False, error_handling=True):
        '''
        Initialize the disambiguation model and Solr connection.
        '''
        if model == 'train':
            self.model = models.Model()
        elif model == 'svm':
            self.model = models.LinearSVM()
        elif model == 'nn':
            self.model = models.NeuralNet()
        elif model == 'bnn':
            self.model = models.BranchingNeuralNet()
        else:
            self.model = models.NeuralNet()

        self.debug = debug
        self.features = features
        self.candidates = candidates
        self.error_handling = error_handling

        self.solr_connection = solr.SolrConnection(SOLR_URL)