Commits

Slavko Zitnik committed aa72e97

collective test new2

  • Participants
  • Parent commits 6a50606

Comments (0)

Files changed (2)

File src/main/java/si/zitnik/research/iobie/core/collective/classifier/CollectiveClassifier.scala

 import si.zitnik.research.iobie.algorithms.crf.feature.packages.JointIterativeFeatureFunctionPackage
 import si.zitnik.research.iobie.algorithms.crf.stat.{Statistics, FMeasure, MUCStatistics}
 import si.zitnik.research.iobie.core.coreference.clustering.impl.TaggedClusterer
-import si.zitnik.research.iobie.statistics.cluster.{BCubedClusterStatistics, MUCClusterStatistics, ClusterStatistics}
+import si.zitnik.research.iobie.statistics.cluster._
 import scala.collection.JavaConversions._
 
 /**
     println("\tResults MUC:" + FMUC)
     val BCubed = BCubedClusterStatistics.scoreExamples(allRealClusters, allTaggedClusters)
     println("\tResults BCubed: " + BCubed)
+    val CEAFe = CEAFClusterStatistics(CEAFStatisticsType.ENTITY_BASED).scoreExamples(allRealClusters, allTaggedClusters)
+    println("\tResults CEAFe: " + CEAFe)
   }
 
   def evaluateNer(examples: Examples, neLabels: ArrayBuffer[ArrayBuffer[String]]) = {

File src/main/java/si/zitnik/research/iobie/core/collective/test/CollectiveTest.scala

 
     val classifier = new CollectiveLearner(
       ACE2004Data.importTrainData(),
-      BasicIterativeFeatureFunctionPackages.standardNerFFunctions(),
-      BasicIterativeFeatureFunctionPackages.standardRelFFPackages(),
-      BasicIterativeFeatureFunctionPackages.bestACE2004CorefFeatureFunctions, 3).train()
+      FeatureFunctionPackages.standardNerFFunctions(),
+      FeatureFunctionPackages.standardRelFFPackages(),
+      FeatureFunctionPackages.bestACE2004CorefFeatureFunctions, 5).train()
     classifier.classify(ACE2004Data.importTestData())
 
   }