def __init__(self, order=1, fullyConnected=False, nIterations=100, topK=1):
""" Create a new mention finder to find a given list of mention types.
entityTypes = list of mention types to find (e.g. group, outcome)
"""
BaseTokenClassifier.__init__(self, 'mallet', topK)
self.simpleTagger = 'java -Xmx2g -cp ' + self.classpath \
+ ' cc.mallet.fst.SimpleTagger'
self.crfOrder = order
if fullyConnected:
self.connectedOption = 'true'
else:
self.connectedOption = 'false'
self.nIterations = nIterations
def __init__(self, binaryThreshold=0.5):
""" Create a new mention finder to find a given list of mention types.
entityTypes = list of mention types to find (e.g. group, outcome)
"""
BaseTokenClassifier.__init__(self, 'megam')
self.binaryThreshold = binaryThreshold