'model_savedir':
'model' + nt,
'results_savedir':
r'.\results',
'figures_savedir':
r'.\figures',
}
print(params)
# 数据生成对象
gen_data = genData(params)
# 建立传统方法计算图
detector = Detector(params)
OAMP_nodes = detector.OAMP()
MMSE_nodes = detector.MMSE()
ZF_nodes = detector.ZF()
# 建立OAMPNet2计算图
if 'OAMPNet2' in params['simulation_algorithms']:
OAMPNet2 = OAMPNet2(params)
OAMPNet2_nodes = OAMPNet2.creat_graph()
# # 建立OAMPNet计算图
if 'OAMPNet' in params['simulation_algorithms']:
OAMPNet = OAMPNet(params)
OAMPNet_nodes = OAMPNet.creat_graph()
# 建立MMNet计算图
if 'MMNet' in params['simulation_algorithms']:
MMNet = MMNet(params)
