search

SciSharp / SiaNet

An easy to use C# deep learning library with CUDA/OpenCL support
https://scisharp.github.io/SiaNet
MIT License
380 stars 83 forks source link

Model.evaluate error on softmax #17

Closed nikosdim1 closed 6 years ago

nikosdim1 commented 6 years ago

Model.evaluate does not return full softmax array to get the index of the predicted class. It only returns value. eg. in a multiclass prediction model.evaluate, expected is : (0.10, 0,05, 0,85) so the winner is 3d value but it returns only the single value of 0.85

deepakkumar1984 commented 6 years ago

Need to work on this.

deepakkumar1984 commented 6 years ago

Can you please check now if you are getting all array.