Skip to content

Commit 1b18ba2

Browse files
committed
Bug fix for mean over all possible edges
1 parent 61b81db commit 1b18ba2

2 files changed

Lines changed: 2 additions & 1 deletion

File tree

src/main/java/edu/virginia/uvacluster/internal/Graph.java

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -333,6 +333,7 @@ public void trainOn(List<Cluster> clusters) {
333333
nextLevel.addAll(root.getChildren());
334334

335335
for (Child child : getRootFeatures()) {
336+
System.out.println(child.getName());
336337
child.addTo(featureMap.get(child.getName()).number);
337338
}
338339

src/main/java/edu/virginia/uvacluster/internal/feature/FeatureUtil.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ public static List<FeatureSet> parse(Set<String> featureKeys) {
5050

5151
// Check if Mean : weight -- to use alternative mean calculation
5252
// over all possible edges
53-
if (weightFeaturePattern.matcher(featureName).matches())
53+
if (weightFeaturePattern.matcher(featureName).matches() && statName.equals("Mean"))
5454
statName = "mean possible";
5555

5656
if (statsMap.get(featureName) == null)

0 commit comments

Comments
 (0)