from sklearn.datasets import load_breast_cancer # Load dataset data = load_breast_cancer() # Organize our data label_names = data['target_names'] labels = data['target'] feature_names = data['feature_names'] features = data['data'] from sklearn.model_selection import train_test_split # Split our data train, test, train_labels, test_labels = train_test_split(features, labels, test_size=0.33, random_state=42) from sklearn.naive_bayes import GaussianNB from sklearn2pmml import PMMLPipeline nb_pipeline = PMMLPipeline([ ('classifier', GaussianNB()) ]) # # Train our classifier nb_pipeline.fit(train, train_labels) # from sklearn2pmml import sklearn2pmml sklearn2pmml(nb_pipeline, 'nb.pmml', with_repr = True,debug=True)
python 保存为pmml格式
最新推荐文章于 2024-11-26 15:00:55 发布