def deserialize_multinomial_nb(model_dict): model = MultinomialNB(model_dict['params']) model.classes_ = np.array(model_dict['classes_']) model.class_count_ = np.array(model_dict['class_count_']) model.class_log_prior_ = np.array(model_dict['class_log_prior_']) model.feature_count_ = np.array(model_dict['feature_count_']) model.feature_log_prob_ = np.array(model_dict['feature_log_prob_']) return model