onehotEncoder.h

Go to the documentation of this file.

#ifndef COSAN_ONEHOTENCODER_H
#define COSAN_ONEHOTENCODER_H
 
#include <cosan/preprocessing/encoder.h>
 
namespace Cosan {
    template<Numeric NumericType>
    class OnehotEncoder : public Encoder<NumericType> {
    public:
        std::vector<int> getEncoding(int colIdx, const std::string &category)  {
            int categoryCount = this->getCategories()[colIdx].size();
            int ordinal = this->getCategories()[colIdx][category];
            std::vector<int> encoding(categoryCount,0);
            encoding[ordinal] = 1;
            return encoding;
        }
    };
 
}
 
 
#endif //COSAN_ONEHOTENCODER_H