From: <ki...@us...> - 2012-09-20 15:44:31
|
Revision: 3847 http://dl-learner.svn.sourceforge.net/dl-learner/?rev=3847&view=rev Author: kirdie Date: 2012-09-20 15:44:22 +0000 (Thu, 20 Sep 2012) Log Message: ----------- now passing the labels as well to the hmm. Modified Paths: -------------- branches/hmm/components-ext/src/main/java/org/dllearner/algorithm/tbsl/learning/SPARQLTemplateBasedLearner2.java Modified: branches/hmm/components-ext/src/main/java/org/dllearner/algorithm/tbsl/learning/SPARQLTemplateBasedLearner2.java =================================================================== --- branches/hmm/components-ext/src/main/java/org/dllearner/algorithm/tbsl/learning/SPARQLTemplateBasedLearner2.java 2012-09-19 15:15:56 UTC (rev 3846) +++ branches/hmm/components-ext/src/main/java/org/dllearner/algorithm/tbsl/learning/SPARQLTemplateBasedLearner2.java 2012-09-20 15:44:22 UTC (rev 3847) @@ -1,7 +1,9 @@ package org.dllearner.algorithm.tbsl.learning; import hmm.HiddenMarkovModel; +import hmm.ResourceInfo; import java.util.ArrayList; +import java.util.Arrays; import java.util.Collection; import java.util.HashMap; import java.util.HashSet; @@ -511,19 +513,21 @@ Query query = template.getQuery(); double score = 0; - Map<List<String>,Collection<String>> segmentToURIs = new HashMap<List<String>,Collection<String>>(); + Map<List<String>,List<ResourceInfo>> segmentToURIs = new HashMap<List<String>,List<ResourceInfo>>(); for(Slot slot: template.getSlots()) { List<String> segment = new LinkedList<String>(); - segment.add(slot.getWords().get(0)); // TODO: split it up? - - Set<String> uris = new HashSet<String>(); - + segment.addAll(Arrays.asList(slot.getWords().get(0).split("\\s"))); + List<ResourceInfo> resourceInfos = new LinkedList<ResourceInfo>(); + for(IndexResultItem item : getIndexResultItems(slot)) { - uris.add(item.getUri()); + // if this gets used at another place, create a function IndexResultItemToResourceInfo() + ResourceInfo info = new ResourceInfo(); + info.setUri(item.getUri()); + info.setLabel(item.getLabel()); } - segmentToURIs.put(segment,uris); + segmentToURIs.put(segment,resources); } HiddenMarkovModel hmm = new HiddenMarkovModel(); hmm.initialization(); This was sent by the SourceForge.net collaborative development platform, the world's largest Open Source development site. |