From 0d1e0e94a1bd7bebd48d7a5922c8496530da75a6 Mon Sep 17 00:00:00 2001 From: Gene Date: Wed, 8 Nov 2023 20:53:40 +0800 Subject: [PATCH 1/2] [MNT] change geatpy to local import --- learnware/reuse/ensemble_pruning.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/learnware/reuse/ensemble_pruning.py b/learnware/reuse/ensemble_pruning.py index d20664c..ab86c66 100644 --- a/learnware/reuse/ensemble_pruning.py +++ b/learnware/reuse/ensemble_pruning.py @@ -1,8 +1,6 @@ import torch import random import numpy as np -import geatpy as ea - from typing import List from learnware.learnware import Learnware @@ -55,6 +53,8 @@ class EnsemblePruningReuser(BaseReuser): np.ndarray Binary one-dimensional vector, 1 indicates that the corresponding model is selected. """ + import geatpy as ea + model_num = v_predict.shape[1] @ea.Problem.single @@ -139,6 +139,8 @@ class EnsemblePruningReuser(BaseReuser): np.ndarray Binary one-dimensional vector, 1 indicates that the corresponding model is selected. """ + import geatpy as ea + model_num = v_predict.shape[1] def find_top_two_freq(row): @@ -253,6 +255,8 @@ class EnsemblePruningReuser(BaseReuser): np.ndarray Binary one-dimensional vector, 1 indicates that the corresponding model is selected. """ + import geatpy as ea + model_num = v_predict.shape[1] v_predict[v_predict == 0.0] = -1 v_true[v_true == 0.0] = -1 From 35188f9184e45e887e3aeeb40d256764026d0a7b Mon Sep 17 00:00:00 2001 From: Gene Date: Wed, 8 Nov 2023 20:54:47 +0800 Subject: [PATCH 2/2] [MNT] remove geatpy and matplotlib in dependencies --- setup.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/setup.py b/setup.py index 67f7254..5d48bfb 100644 --- a/setup.py +++ b/setup.py @@ -54,7 +54,6 @@ REQUIRED = [ "numpy>=1.20.0", "pandas>=0.25.1", "scipy>=1.0.0", - "matplotlib>=3.1.3", "torch>=1.11.0", "cvxopt>=1.3.0", "tqdm>=4.65.0", @@ -67,7 +66,6 @@ REQUIRED = [ "torchvision>=0.15.1", "sqlalchemy>=2.0.21", "shortuuid>=1.0.11", - "geatpy>=2.7.0", "docker>=6.1.3", "rapidfuzz>=3.4.0", "torchtext>=0.16.0",