ordinalEncoder.h

Go to the documentation of this file.

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