From 7334cb060c86ae454f5c1e26571f4f46b2f385c7 Mon Sep 17 00:00:00 2001 From: Ashok Arora Date: Thu, 1 Oct 2020 13:29:25 +0530 Subject: [PATCH] Skeleton for Label Encoder --- include/preprocessing.hpp | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/include/preprocessing.hpp b/include/preprocessing.hpp index bc2a0fe..d842c03 100644 --- a/include/preprocessing.hpp +++ b/include/preprocessing.hpp @@ -25,4 +25,16 @@ Matrix Preprocessing::normalize(Matrix mat, std::string dim) { return result; } -#endif /* _preprocessing_h_ */ \ No newline at end of file +class LabelEncoder { + private: + Matrix labels; + + public: + void fit(Matrix Y); + Matrix fit_transform(Matrix Y); + Matrix get_params(); + void set_params(); + Matrix transform(Matrix Y); +} + +#endif /* _preprocessing_h_ */