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

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')

源码文件结构

img

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

以下答案由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' 这样的错误了。
如果我的回答解决了您的问题,请采纳!

可能修改没保存,刷新或者重新打开试试?

该回答引用chatgpt:
可能的解决方法如下:

1检查是否已经正确安装了flearn和其它依赖包。你可以在项目根目录下运行pip install -r requirements.txt来安装所需的依赖包。
2检查flearn/models/mnist目录下是否存在mclr.py模块,或者检查模块名是否正确。如果不存在,请从仓库的原始代码中获取这个模块并将其复制到正确的位置。
3检查flearn/models/mnist目录是否有一个__init__.py文件。这个文件用来标记一个目录为Python包,如果这个文件不存在,Python可能无法正确地加载这个包。
4检查flearn/models/mnist目录是否在Python的搜索路径中。你可以使用以下代码来检查Python的搜索路径:import sys; print(sys.path)。

如果你确定上面回答的原因都排除了,那清除一下缓存吧,再运行,说不定就OK了。

python -m compileall .

No module named ‘models‘ 解决方法(亲测)

可以借鉴下
https://blog.csdn.net/OnlyHeart_/article/details/124935824