Shuli Gao JackerGao
Kukanani
/ save_and_load_sklearn_gmm
Created
April 10, 2018 22:49
Save and load a Scikit-learn GMM to file using np.save
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import numpy as np | |
| from sklearn import mixture | |
| # make a GMM | |
| gmm = mixture.GaussianMixture(n_components=n_components, covariance_type=cv_type) | |
| # fit the GMM however you'd like | |
| # save to file | |
| gmm_name = 'new_gmm' | |
| np.save(gmm_name + '_weights', gmm.weights_, allow_pickle=False) |