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メソッドと同等です。