public static void getWordAnology(
RawSemanticSpace space, String word1, String word2, String word3) {
if (!(space.contains(word1) && space.contains(word2) && space.contains(word3))) {
System.out.println("at least one word is not in the space");
return;
}
double[] v2 = RealVector.norm(space.getVector(word2).getMatrix().getData());
double[] v1 = RealVector.norm(space.getVector(word1).getMatrix().getData());
double[] v3 = RealVector.norm(space.getVector(word3).getMatrix().getData());
SimpleMatrix v4 =
new SimpleMatrix(1, v1.length, true, RealVector.add(RealVector.subtract(v2, v1), v3));
Neighbor[] neighbors;
neighbors = space.getNeighbors(v4, 20, new String[] {word1, word2, word3});
System.out.println("\"" + word1 + "\" to \"" + word2 + "\" is like \"" + word3 + "\" to: ");
for (int i = 0; i < neighbors.length; i++) {
System.out.println(neighbors[i].word + " " + neighbors[i].sim);
}
}
public static String predictWordAnology(
RawSemanticSpace space, String word1, String word2, String word3) {
if (!(space.contains(word1) && space.contains(word2) && space.contains(word3))) {
System.out.println("at least one word is not in the space");
return "";
}
double[] v2 = RealVector.norm(space.getVector(word2).getMatrix().getData());
double[] v1 = RealVector.norm(space.getVector(word1).getMatrix().getData());
double[] v3 = RealVector.norm(space.getVector(word3).getMatrix().getData());
SimpleMatrix v4 =
new SimpleMatrix(1, v1.length, true, RealVector.add(RealVector.subtract(v2, v1), v3));
Neighbor[] neighbors;
neighbors = space.getNeighbors(v4, 20, new String[] {word1, word2, word3});
for (int i = 0; i < neighbors.length; i++) {
String neighbor = neighbors[i].word;
if (!(word1.equals(neighbor) || word2.equals(neighbor) || word3.equals(neighbor)))
return neighbor;
}
return "";
}