not good but great

プログラミング、アート、映画・本の感想について書きます。

2017-04-13から1日間の記事一覧

numpyで値が大きい上位K件の配列インデックスを取得

gist.github.com argmaxだと最大値のインデックスしか取れない。上位k件のインデックスが欲しかった。調べて見るとnumpy argpartitionというのがあるらしい。それを使って書いた。 MNISTで nearest neighbor / int か float か,それが問題です / np.argpa…