示例#1
0
def main(goldFile, predFile, outCSVFile, description, fieldName1, fieldValue1):
    (goldRoot, goldNodes) = evaluate.getCoarseGrainedTreeLabelsFile(goldFile)
    (predRoot, predNodes) = evaluate.getCoarseGrainedTreeLabelsFile(predFile)
    rootData = evaluate.printStatsCoarseInt(goldRoot, predRoot, prefix='root')
    nodeData = evaluate.printStatsCoarseInt(goldNodes, predNodes)
    nodeData.update(rootData)
    if description:
        nodeData = evaluate.ins(['Description'],
                                [description], nodeData)
    if fieldName1 and fieldValue1:
        nodeData = evaluate.ins([fieldName1],
                                [fieldValue1], nodeData)

    evaluate.statsToFile(nodeData, outCSVFile, delim=';')
示例#2
0
 def test_tree_reading(self):
     (root, node) = evaluate.getCoarseGrainedTreeLabelsFile(
         'tests/ptb/small_treebank.ptb')
     expectedRoot = [NEU, NEG, NEU, NEU]
     expectedLabels = [NEU, NEG, POS, POS, POS, NEG, POS]
     expectedLabels += [NEG, NEU, POS, POS, POS, NEU, NEG]
     expectedLabels += [NEU, NEG, POS, POS, POS, NEG, POS]
     expectedLabels += [NEU, NEG, POS, POS, POS, NEG, POS]
     self.assertSequenceEqual(root, expectedRoot)
     self.assertSequenceEqual(node, expectedLabels)