TadaoYamaokaの開発日記

個人開発しているスマホアプリや将棋AIの開発ネタを中心に書いていきます。

続:どの駒が評価値に寄与しているかを可視化する

先日記事にした駒の働きを可視化するツール(feature_importance.py)に、SVGを出力する機能を追加した。

前回は数値で出力して、Excelの条件付き書式のカラースケールで、色に変換してから、ShogiGUIに画像として重ね合わせるということを行っていたが、手間がかかるので、SVG画像を出力することで加工なしで確認できるようにした。

ソースコード

前回はGitHub Gistで公開していたが、dlshogiのutilsに追加した。
DeepLearningShogi/feature_importance.py at master · TadaoYamaoka/DeepLearningShogi · GitHub

実行例

オプション「--svg」にSVGファイルのパスを指定する。

python -m dlshogi.utils.feature_importance model.onnx "l2r3nl/1P1+Rsk3/2nPppgp1/p5p1p/1pP2P3/PSp1s1P1P/4P4/1KG2G3/LN6L b 2B2Pgsn 85" --svg a.svg

実行には、cshogiの最新版(v0.4.9)が必要。

表示結果

SVG画像は、ブラウザで開いて確認できる。