ml5.jsで複数のクラスを学習させても二クラスしか学習されない

複数のクラスからなる画像群をaddImageしていたったのに、
classify/predictさせて見せると必ず初めの二つのクラスのどっちかを答えるようになる…

つまり、A,B,C,Dのクラスを覚えさせるように与えてもA,Bしか使われない.

以下にそのIssueが上がってました。

https://github.com/ml5js/ml5-library/issues/164

クラス数を指定すればとりあえず動くとのこと

えぇって感じ…

 

featureExtractor.numClasses = クラス数

を指定すればちゃんと2クラス以上も識別される.

このときに初めて他のOptionも知った.

https://ml5js.org/docs/FeatureExtractor に書いてある.

hiddenUnits : 隠れ層の数

epochs : エポック数. 学習反復回数みたいなもの

他にはbatchSize, learningRate等々もある.

それでもやはりシンプルに本当に必要そうなパラメータだけだ…

 

About the author

コメントを残す