17.3.12 モデル・パフォーマンスの評価
evaluate
コンビニエンス・メソッドを使用して、モデルの様々なメトリックを評価できます:
opg4j> model.evaluate(fullGraph, testEdges).print()
model.evaluate(fullGraph,testEdges).print();
model.evaluate(full_graph,test_edges).print()
ラベルの推測と同様に、タスクが分類タスクの場合、決定しきい値を追加パラメータとして追加できます。
opg4j> model.evaluate(fullGraph, testEdges, 6f).print()
model.evaluate(fullGraph,testEdges, 6f).print();
model.evaluate(full_graph,test_edges, 6).print()
分類モデルの場合、出力は次のようになります:
+------------------------------------------+
| Accuracy | Precision | Recall | F1-Score |
+------------------------------------------+
| 0.8488 | 0.8523 | 0.831 | 0.8367 |
+------------------------------------------+
回帰モデルの場合、出力は次のようになります:
+--------------------+
| MSE |
+--------------------+
| 0.9573243436116953 |
+--------------------+
分類モデルの場合、evaluateLabels
メソッドも使用可能で、これはevaluate
メソッドと同等です。