No module named 'flearn.models.mnist.mclr.py'; 'flearn.models.mnist.mclr' is not a package

Question

FedProx源码运行报错：No module named 'flearn.models.mnist.mclr.py'; 'flearn.models.mnist.mclr' is not a package

在pycharm上运行，对main.py进行运行，报错No module named 'flearn.models.nist.stacked_lstm' ，修改dataset默认值为mnist

def read_options():
    ''' Parse command line arguments or load defaults '''
    parser = argparse.ArgumentParser()

    parser.add_argument('--optimizer',
                        help='name of optimizer;',
                        type=str,
                        choices=OPTIMIZERS,
                        default='fedavg')
    parser.add_argument('--dataset',
                        help='name of dataset;',
                        type=str,
                        choices=DATASETS,
                        default='mnist')
    parser.add_argument('--model',
                        help='name of model;',
                        type=str,
                        default='mclr.py')

仍然报错ModuleNotFoundError: No module named 'flearn.models.mnist.mclr.py'; 'flearn.models.mnist.mclr' is not a package

引入该数据包相关代码：

 # load selected model
    if parsed['dataset'].startswith("synthetic"):  # all synthetic datasets use the same model
        model_path = '%s.%s.%s.%s' % ('flearn', 'models', 'synthetic', parsed['model'])
    else:
        model_path = '%s.%s.%s.%s' % ('flearn', 'models', parsed['dataset'], parsed['model'])

    mod = importlib.import_module(model_path)
    learner = getattr(mod, 'Model')

源码文件结构

源码地址：https://github.com/litian96/FedProx

Answer 1

以下答案由GPT-3.5大模型与博主波罗歌共同编写：
根据错误提示，可以看出 flearn.models.mnist.mclr 并不是一个包，所以无法直接引入 mclr.py。因此需要做如下修改：

在 read_options() 函数中将 model 参数默认值改为 "mclr" 即可，因为 "mclr" 已经在 flearn.models.mnist 模块里了，不需要再加上 .py。

修改后的代码如下：

def read_options():
    ''' Parse command line arguments or load defaults '''
    parser = argparse.ArgumentParser()

    parser.add_argument('--optimizer',
                        help='name of optimizer;',
                        type=str,
                        choices=OPTIMIZERS,
                        default='fedavg')
    parser.add_argument('--dataset',
                        help='name of dataset;',
                        type=str,
                        choices=DATASETS,
                        default='mnist')
    parser.add_argument('--model',
                        help='name of model;',
                        type=str,
                        default='mclr')  # 默认值改为 mclr，而不是 mclr.py

    return parser.parse_args()

另外，在加载模型时，需要注意修改模型的路径，例如：

# load selected model
if parsed['dataset'].startswith("synthetic"):
    model_path = '%s.%s.%s.%s' % ('flearn', 'models', 'synthetic', parsed['model'])
else:
    model_path = '%s.%s.%s' % ('flearn', 'models', parsed['model'])

mod = importlib.import_module(model_path)
learner = getattr(mod, 'Model')

这样就可以避免出现 No module named 'flearn.models.mnist.mclr.py' 这样的错误了。
如果我的回答解决了您的问题，请采纳！