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