How do I resolve "KeyError: 'brand'" when running an experiment using Azure Automated Machine Learning?

Question

I am using the Azure Automated Machine Learning SDK to train a machine learning model on my dataset. However, after the experiment, all my training iterations fail with a KeyError: 'brand' error even if the model training itself succeeded.

How can I resolve this?

Answer 1

If a new environment was created after 10 June 2020 using SDK 1.7.0 or lower, training may fail with the above error due to an update in the py-cpuinfo package. (Environments created on or before 10 June 2020 are unaffected, as well as experiments run on remote compute as cached training images are used.) To work around this issue, either of the two following steps can be taken:

1. Update the SDK version to 1.8.0 or higher (this will also downgrade py-cpuinfo to 5.0.0):

pip install --upgrade azureml-sdk[automl]

2. Downgrade the installed version of py-cpuinfo to 5.0.0:

pip install py-cpuinfo==5.0.0