def __init__(self): with open(os.path.join(os.path.dirname(__file__), 'Config.json'), 'r') as json_file: config = json.load(json_file) self._shots = dict() self._shots['train'] = config['shots']['train'] self._shots['test'] = config['shots']['test'] self._shots['val'] = config['shots']['test'] self._tags = config['diagnosis']['tags'] # 截取信号的起始时间(ms) self._start_time = config['diagnosis']['start_time'] # 重采样的信号频率(kHz) self._sample_rate = config['diagnosis']['sample_rate'] self._normalized = config['normalized'] self._directory = config['directory'] if not os.path.exists(self._directory): os.makedirs(self._directory) # 复制配置文件到输出路径 with open(os.path.join(self._directory, 'Config.json'), 'w') as json_file: json.dump(config, json_file, indent=4) exit() ddb = Query() if self._normalized: self._normalize_param = ddb.get_normalize_parm(self._tags)
def __init__(self, normalized=False): config = configparser.ConfigParser() config.read( os.path.join(os.path.dirname(__file__), 'DataSetConfig.ini')) self._tags = ast.literal_eval(config['Diagnosis']['tags']) self._sample_rate = int(config['Diagnosis']['sample_rate']) self._frame_size = int(config['Diagnosis']['frame_size']) self._step = int(config['Diagnosis']['step']) self._npy_path = config['path']['npy'] if not os.path.exists(self._npy_path): os.makedirs(self._npy_path) ddb = Query() self._normalized = normalized if normalized: self._normalize_param = ddb.get_normalize_parm(self._tags) my_query = {'IsValidShot': True, 'IsDisrupt': False} self._shots = ddb.query(my_query) my_query = { 'IsValidShot': True, 'IsDisrupt': True, 'CqTime': { "$gte": 0.05 }, 'IpFlat': { '$gte': 110 } } self._shots += ddb.query(my_query)