Unverified Commit e21a6984 authored by liuzhe-lz's avatar liuzhe-lz Committed by GitHub
Browse files

[v2.0] Refactor code hierarchy (part 2) (#2987)

parent f98ee672
...@@ -11,7 +11,7 @@ import numpy ...@@ -11,7 +11,7 @@ import numpy
from scipy.stats import norm from scipy.stats import norm
from scipy.optimize import minimize from scipy.optimize import minimize
import nni.metis_tuner.lib_data as lib_data from . import lib_data
def next_hyperparameter_expected_improvement(fun_prediction, def next_hyperparameter_expected_improvement(fun_prediction,
......
...@@ -15,14 +15,14 @@ import numpy as np ...@@ -15,14 +15,14 @@ import numpy as np
from schema import Schema, Optional from schema import Schema, Optional
from nni import ClassArgsValidator from nni import ClassArgsValidator
import nni.metis_tuner.lib_constraint_summation as lib_constraint_summation from . import lib_constraint_summation
import nni.metis_tuner.lib_data as lib_data from . import lib_data
import nni.metis_tuner.Regression_GMM.CreateModel as gmm_create_model from .Regression_GMM import CreateModel as gmm_create_model
import nni.metis_tuner.Regression_GMM.Selection as gmm_selection from .Regression_GMM import Selection as gmm_selection
import nni.metis_tuner.Regression_GP.CreateModel as gp_create_model from .Regression_GP import CreateModel as gp_create_model
import nni.metis_tuner.Regression_GP.OutlierDetection as gp_outlier_detection from .Regression_GP import OutlierDetection as gp_outlier_detection
import nni.metis_tuner.Regression_GP.Prediction as gp_prediction from .Regression_GP import Prediction as gp_prediction
import nni.metis_tuner.Regression_GP.Selection as gp_selection from .Regression_GP import Selection as gp_selection
from nni.tuner import Tuner from nni.tuner import Tuner
from nni.utils import OptimizeMode, extract_scalar_reward from nni.utils import OptimizeMode, extract_scalar_reward
......
...@@ -13,9 +13,9 @@ from scipy.optimize import linear_sum_assignment ...@@ -13,9 +13,9 @@ from scipy.optimize import linear_sum_assignment
from sklearn.metrics.pairwise import rbf_kernel from sklearn.metrics.pairwise import rbf_kernel
from nni.utils import OptimizeMode from nni.utils import OptimizeMode
from nni.networkmorphism_tuner.graph_transformer import transform from .graph_transformer import transform
from nni.networkmorphism_tuner.utils import Constant from .utils import Constant
from nni.networkmorphism_tuner.layers import is_layer from .layers import is_layer
def layer_distance(a, b): def layer_distance(a, b):
......
...@@ -9,7 +9,7 @@ from queue import Queue ...@@ -9,7 +9,7 @@ from queue import Queue
import numpy as np import numpy as np
import torch import torch
from nni.networkmorphism_tuner.layer_transformer import ( from .layer_transformer import (
add_noise, add_noise,
wider_bn, wider_bn,
wider_next_conv, wider_next_conv,
...@@ -20,7 +20,7 @@ from nni.networkmorphism_tuner.layer_transformer import ( ...@@ -20,7 +20,7 @@ from nni.networkmorphism_tuner.layer_transformer import (
init_conv_weight, init_conv_weight,
init_bn_weight, init_bn_weight,
) )
from nni.networkmorphism_tuner.layers import ( from .layers import (
StubAdd, StubAdd,
StubConcatenate, StubConcatenate,
StubReLU, StubReLU,
...@@ -36,7 +36,7 @@ from nni.networkmorphism_tuner.layers import ( ...@@ -36,7 +36,7 @@ from nni.networkmorphism_tuner.layers import (
layer_description_extractor, layer_description_extractor,
layer_description_builder, layer_description_builder,
) )
from nni.networkmorphism_tuner.utils import Constant from .utils import Constant
class NetworkDescriptor: class NetworkDescriptor:
......
...@@ -5,8 +5,8 @@ from copy import deepcopy ...@@ -5,8 +5,8 @@ from copy import deepcopy
from random import randrange, sample from random import randrange, sample
from nni.networkmorphism_tuner.graph import NetworkDescriptor from .graph import NetworkDescriptor
from nni.networkmorphism_tuner.layers import ( from .layers import (
StubDense, StubDense,
StubReLU, StubReLU,
get_batch_norm_class, get_batch_norm_class,
...@@ -15,7 +15,7 @@ from nni.networkmorphism_tuner.layers import ( ...@@ -15,7 +15,7 @@ from nni.networkmorphism_tuner.layers import (
get_pooling_class, get_pooling_class,
is_layer, is_layer,
) )
from nni.networkmorphism_tuner.utils import Constant from .utils import Constant
def to_wider_graph(graph): def to_wider_graph(graph):
......
...@@ -3,7 +3,7 @@ ...@@ -3,7 +3,7 @@
import numpy as np import numpy as np
from nni.networkmorphism_tuner.layers import ( from .layers import (
StubDense, StubDense,
StubReLU, StubReLU,
get_batch_norm_class, get_batch_norm_class,
......
...@@ -7,7 +7,7 @@ from collections.abc import Iterable ...@@ -7,7 +7,7 @@ from collections.abc import Iterable
import torch import torch
from torch import nn from torch import nn
from torch.nn import functional from torch.nn import functional
from nni.networkmorphism_tuner.utils import Constant from .utils import Constant
class AvgPool(nn.Module): class AvgPool(nn.Module):
......
...@@ -10,10 +10,10 @@ import os ...@@ -10,10 +10,10 @@ import os
from schema import Optional, Schema from schema import Optional, Schema
from nni.tuner import Tuner from nni.tuner import Tuner
from nni.utils import OptimizeMode, extract_scalar_reward from nni.utils import OptimizeMode, extract_scalar_reward
from nni.networkmorphism_tuner.bayesian import BayesianOptimizer from .bayesian import BayesianOptimizer
from nni.networkmorphism_tuner.nn import CnnGenerator, MlpGenerator from .nn import CnnGenerator, MlpGenerator
from nni.networkmorphism_tuner.utils import Constant from .utils import Constant
from nni.networkmorphism_tuner.graph import graph_to_json, json_to_graph from .graph import graph_to_json, json_to_graph
from nni import ClassArgsValidator from nni import ClassArgsValidator
logger = logging.getLogger("NetworkMorphism_AutoML") logger = logging.getLogger("NetworkMorphism_AutoML")
......
...@@ -3,14 +3,14 @@ ...@@ -3,14 +3,14 @@
from abc import abstractmethod from abc import abstractmethod
from nni.networkmorphism_tuner.graph import Graph from .graph import Graph
from nni.networkmorphism_tuner.layers import (StubDense, StubDropout1d, from .layers import (StubDense, StubDropout1d,
StubReLU, get_batch_norm_class, StubReLU, get_batch_norm_class,
get_conv_class, get_conv_class,
get_dropout_class, get_dropout_class,
get_global_avg_pooling_class, get_global_avg_pooling_class,
get_pooling_class) get_pooling_class)
from nni.networkmorphism_tuner.utils import Constant from .utils import Constant
class NetworkGenerator: class NetworkGenerator:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment