Example #1
0
def loadDico(components_,
             transform_algorithm='lars',
             kwargs={'transform_n_nonzero_coefs': 5}):
    n_components = components_.shape[0]
    dico = MiniBatchDictionaryLearning(n_components=n_components,
                                       alpha=1,
                                       n_iter=500)
    dico.set_params(transform_algorithm=transform_algorithm, **kwargs)
    dico.components_ = components_
    dico.n_components_ = n_components
    return dico