def deviation_network(self, input_shape, network_depth, modelname, testflag):
        '''
        construct the deviation network-based detection model
        '''
        if network_depth == 4:
            model = self.dev_network_d(input_shape, modelname, testflag)
        elif network_depth == 2:
            model = self.auto_encoder(input_shape)

        else:
            sys.exit("The network depth is not set properly")
        return model

        construct the deviation network-based detection model
        
adbench/baseline/FEAWAD/run.py