Skip to content
Snippets Groups Projects
Commit 2d042faa authored by Eike Cochu's avatar Eike Cochu
Browse files

added relevance change to models

parent c2aef213
No related branches found
No related tags found
No related merge requests found
......@@ -224,6 +224,7 @@ public class DTMAnalyzer extends Analyzer {
final double[] relevances = new double[sequencesCount];
double relevanceSum = 0;
double prevRelevance = 0;
// for each sequence
for (int idxSeq = 0, sequenceOffset = 0; idxSeq < sequencesCount; idxSeq++) {
......@@ -266,6 +267,7 @@ public class DTMAnalyzer extends Analyzer {
newSequenceFull.setWindow(newWindows.get(idxSeq));
newSequenceFull.setWords(newSeqTopicWords);
newSequenceFull.setRelevance(relevance);
newSequenceFull.setRelevanceChange(relevance - prevRelevance);
newSequenceFull.setTopic(new Topic(newTopic.getId()));
newSequences.add(newSequenceFull);
newTopicSequences.add(new Sequence(newSequenceFull.getId()));
......@@ -273,6 +275,7 @@ public class DTMAnalyzer extends Analyzer {
sequenceOffset += sequenceSize;
relevanceSum += relevance;
relevances[idxSeq] = relevance;
prevRelevance = relevance;
}
// sort topic words and generate topic name
......
......@@ -19,6 +19,8 @@ public class Sequence implements Model<ObjectId>, Comparable<Sequence>, Serializ
private Double relevance;
private Double relevanceChange;
public Sequence() {}
public Sequence(final ObjectId id) {
......@@ -51,6 +53,14 @@ public class Sequence implements Model<ObjectId>, Comparable<Sequence>, Serializ
this.relevance = relevance;
}
public Double getRelevanceChange() {
return relevanceChange;
}
public void setRelevanceChange(Double relevanceChange) {
this.relevanceChange = relevanceChange;
}
@Override
public int compareTo(final Sequence o) {
return window.getId() - o.getWindow().getId();
......@@ -58,7 +68,8 @@ public class Sequence implements Model<ObjectId>, Comparable<Sequence>, Serializ
@Override
public String toString() {
return "Sequence [id=" + id + ", window=" + window + ", relevance=" + relevance + "]";
return "Sequence [id=" + id + ", window=" + window + ", relevance=" + relevance + ", relevanceChange="
+ relevanceChange + "]";
}
}
......@@ -23,6 +23,8 @@ public class SequenceFull implements Model<ObjectId>, Comparable<SequenceFull>,
private Double relevance;
private Double relevanceChange;
@Reference
@QueryIgnore(multi = true)
private Topic topic;
......@@ -57,6 +59,14 @@ public class SequenceFull implements Model<ObjectId>, Comparable<SequenceFull>,
this.relevance = relevance;
}
public Double getRelevanceChange() {
return relevanceChange;
}
public void setRelevanceChange(Double relevanceChange) {
this.relevanceChange = relevanceChange;
}
public Topic getTopic() {
return topic;
}
......@@ -80,8 +90,8 @@ public class SequenceFull implements Model<ObjectId>, Comparable<SequenceFull>,
@Override
public String toString() {
return "SequenceFull [id=" + id + ", window=" + window + ", relevance=" + relevance + ", topic=" + topic
+ ", words=" + words + "]";
return "SequenceFull [id=" + id + ", window=" + window + ", relevance=" + relevance + ", relevanceChange="
+ relevanceChange + ", topic=" + topic + ", words=" + words + "]";
}
}
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment