From b8653d54efe2a8c94ae408c0c2da9bdd0b43ecdd Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Tue, 29 Dec 2020 20:38:18 +0800
Subject: Encode class names to label and some access improvement

1. Encode class names using LabelEncoder from sklearn
2. Remove unneeded class variables
3. Protect some variables from being accessed in userspace
---
 requirements.txt | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'requirements.txt')

diff --git a/requirements.txt b/requirements.txt
index 765977f..7cf8570 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,4 +2,5 @@ torch~=1.7.1
 torchvision~=0.8.0a0+ecf4e9c
 numpy~=1.19.4
 tqdm~=4.55.0
-Pillow~=8.0.1
\ No newline at end of file
+Pillow~=8.0.1
+scikit-learn~=0.23.2
\ No newline at end of file
-- 
cgit v1.2.3