Commits

Slavko Zitnik committed 17eabbb

IOBIE full implementation

  • Participants
  • Parent commits 693f65f

Comments (0)

Files changed (1)

File src/main/java/si/zitnik/research/iobie/core/collective/learner/CollectiveLearner.scala

       logger.info("Iteration " + i)
 
       //NER
+      println("A")
       neClassifiers(i) = new NERMentionLearner(examples, new ArrayList(neFeatureFunctions ++ iterativeAndOntologyFeatureFunctions), Label.NE, "NECollective_%d".format(i)).train()
 
       //REL
+      println("B")
       relClassifiers(i) = new RelationshipMultipleLearner(examples, new ArrayList(relFeatureFunctions ++ iterativeAndOntologyFeatureFunctions), Label.REL, Array(0,1,2,3), "RECollective_%d_".format(i)).train()
 
       //COREF
+      println("C")
       corefClassifiers(i) = new CorefMultipleLearner(mentionExamples, new ArrayList(corefFeatureFunctions ++ iterativeAndOntologyFeatureFunctions), modelSaveFilename = "COREFCollective_%d_".format(i)).train()
 
       if (i+1 < numOfIters) { //is not the last step