17.2.10 Supervised GraphWiseモデルの頂点ラベルの推測

次のコードに示すように、任意のグラフ(トレーニング中に表示されなかった頂点またはグラフを含む)の頂点のラベルを推測できます。

opg4j> var labels = model.inferLabels(fullGraph, testVertices)
opg4j> labels.head().print()
PgxFrame labels = model.inferLabels(fullGraph,testVertices);
labels.head().print();
labels = model.infer_labels(full_graph, test_vertices)
labels.print()
出力は、次の出力例のようになります。

+----------------------------------+
| vertexId | label                 |
+----------------------------------+
| 2        | Neural Networks       |
| 6        | Theory                |
| 7        | Case Based            |
| 22       | Rule Learning         |
| 30       | Theory                |
| 34       | Neural Networks       |
| 47       | Case Based            |
| 48       | Probabalistic Methods |
| 50       | Theory                |
| 52       | Theory                |
+----------------------------------+

同様に、次のコードに示すように、予測ロジットを推測することで、各クラスのモデルの信頼度を取得することもできます。

opg4j> var logits = model.inferLogits(fullGraph, testVertices)
opg4j> labels.head().print()
PgxFrame logits = model.inferLogits(fullGraph,testVertices);
logits.head().print();
logits = model.infer_logits(full_graph, test_vertices)
logits.print()