ludwig Image Classification: 配置

l7mqbcuq  于 5个月前  发布在  其他
关注(0)|答案(4)|浏览(62)

配置文件内容如下:

config = {
    'input_features': [
        {
            'name': 'image_path',
            'type': 'image',
            'preprocessing': {'num_processes': 4},
            'encoder': 'stacked_cnn',
            'conv_layers': [
                {'num_filters': 32, 'filter_size': 3, 'pool_size': 2, 'pool_stride': 2},
                {'num_filters': 64, 'filter_size': 3, 'pool_size': 2, 'pool_stride': 2, 'dropout': 0.4}
            ],
            'fc_layers': [{'output_size': 128, 'dropout': 0.4}]
        }
    ],
    'output_features': [{'name': 'label', 'type': 'binary'}],
    'trainer': {'epochs': 5}
}
j8yoct9x

j8yoct9x1#

你好@AnaMiguelRodrigues1,感谢你提出这个问题和建议的解决方法!
我只是想了解一些事情:

  1. 你在运行哪个版本的Ludwig?
  2. 你看到的错误是什么?
wixjitnu

wixjitnu2#

我正在使用最新版本,ludwig==0.8.dev0。错误是语法错误,只是在配置中'encoder'字段没有正确定义。

rmbxnbpk

rmbxnbpk3#

明白了!您是否愿意将修复贡献给Ludwig?

xriantvc

xriantvc4#

我很想纠正它,但在Ludwig的官方网站上(https://ludwig.ai/latest/examples/mnist/),我没有权限更改网站和纠正配置。

相关问题