diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2020-12-29 20:38:18 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2020-12-29 20:38:18 +0800 |
| commit | b8653d54efe2a8c94ae408c0c2da9bdd0b43ecdd (patch) | |
| tree | 0aca443c5f2b0387fae48aa43611ca92d6015bbe /requirements.txt | |
| parent | 6e94fdb587656074dc2e65a80e51b8446f834b41 (diff) | |
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
Diffstat (limited to 'requirements.txt')
| -rw-r--r-- | requirements.txt | 3 |
1 files changed, 2 insertions, 1 deletions
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 |