def deserialize_gaussian_nb(model_dict): model = GaussianNB(model_dict['params']) model.classes_ = np.array(model_dict['classes_']) model.class_count_ = np.array(model_dict['class_count_']) model.class_prior_ = np.array(model_dict['class_prior_']) model.theta_ = np.array(model_dict['theta_']) model.sigma_ = np.array(model_dict['sigma_']) model.epsilon_ = model_dict['epsilon_'] return model