SoftmaxModel.h

Go to the documentation of this file.

#ifndef GRT_SOFTMAX_MODEL_HEADER
#define GRT_SOFTMAX_MODEL_HEADER

#include "../../CoreModules/Classifier.h"

GRT_BEGIN_NAMESPACE
 
class SoftmaxModel{
public:
 SoftmaxModel(){
 classLabel = 0;
 N = 0;
 }
 ~SoftmaxModel(){
 
 }
 
 bool init(UINT classLabel,UINT N){
 this->classLabel = classLabel;
 this->N = N;
 
 //Resize the weights
 w.clear();
 w.resize(N);
 
 //Randomize the weights
 Random rand;
 for(UINT i=0; i<N; i++){
 w[i] = rand.getRandomNumberUniform(-0.1,0.1);
 }
 w0 = rand.getRandomNumberUniform(-0.1,0.1);
 
 return true;
 }
 
 Float compute(const VectorFloat &x){
 Float sum = w0;
 for(UINT i=0; i<N; i++){
 sum += x[i]*w[i];
 }
 return 1.0 / (1.0+exp(-sum));
 }
 
 UINT classLabel;
 UINT N; //The number of dimensions
 VectorFloat w; //The coefficents
 Float w0;
 
};

GRT_END_NAMESPACE

#endif //SoftmaxModel