Python: list iteration возвращает только последнее значение

Question

Я использую scikit-learn для обучения GMM и пытаюсь изменить количество компонентов смеси, перейдя по списку целых чисел. Но когда я печатаю мои результирующие модели, я получаю только те, у которых есть 3 компонента смеси, или все, что я ставлю в качестве последнего элемента в моем списке.

Это мой код:

from sklearn.mixture import GMM 

class_names = ['name1','name2','name3'] 
covs = ['spherical', 'diagonal', 'tied', 'full'] 
num_comp = [1,2,3] 

models = {} 
for c in class_names: 
    models[c] = dict((covar_type,GMM(n_components=num, 
       covariance_type=covar_type, init_params='wmc',n_init=1, n_iter=10)) for covar_type in covs for num in num_comp) 
print models 

Может кто-то помочь, пожалуйста? Большое спасибо заранее!

Answer 1

Это происходит потому, что в выражении:

dict((covar_type,GMM(n_components=num, 
       covariance_type=covar_type, init_params='wmc',n_init=1, n_iter=10)) for covar_type in covs for num in num_comp) 

Вы используете один и тот же covar_type как ключ в течение всех итераций, таким образом переписывая тот же элемент.

Если записать код более понятным способом, это то, что это происходит:

data = dict() 
for covar_type in covs: 
    for num in num_comp: 
     # covar_type is the same for all iterations of this loop 
     # hence only the last one "survives" 
     data[covar_type] = GMM(...) 

Если вы хотите сохранить все значения, которые вы должны использовать список значений вместо одного значения или изменить ключ.

Для списка значений:

data = dict() 
for covar_type in covs: 
    data[covar_type] = values = [] 
    for num in num_comp: 
     values.append(GMM(...)) 

Для различных ключей:

data = dict() 
for covar_type in covs: 
    for num in num_comp: 
     data[(covar_type, num)] = GMM(...)