OpenCV 贝叶斯分类器示例

最新推荐文章于 2022-09-25 18:35:35 发布

zhqh100

最新推荐文章于 2022-09-25 18:35:35 发布

阅读量407

点赞数 2

分类专栏： python

本文链接：https://blog.csdn.net/zhqh100/article/details/108099320

版权

python 专栏收录该内容

51 篇文章 0 订阅

订阅专栏

代码参考 https://github.com/mbeyeler/opencv-machine-learning/blob/master/notebooks/07.01-Implementing-Our-First-Bayesian-Classifier.ipynb

from sklearn import datasets
import numpy as np
from sklearn import model_selection as ms
import cv2
from sklearn import metrics
import matplotlib.pyplot as plt

X, y = datasets.make_blobs(100, 2, centers=2, random_state=1701, cluster_std=2)

plt.style.use('ggplot')

plt.figure(figsize=(10, 6))
plt.scatter(X[:, 0], X[:, 1], c=y, s=50)

X_train, X_test, y_train, y_test = ms.train_test_split(
    X.astype(np.float32), y, test_size=0.1
)

model_norm = cv2.ml.NormalBayesClassifier_create()

model_norm.train(X_train, cv2.ml.ROW_SAMPLE, y_train)
_, y_pred = model_norm.predict(X_test)

metrics.accuracy_score(y_test, y_pred)


def plot_decision_boundary(model, X_test, y_test):
    # create a mesh to plot in
    h = 0.02  # step size in mesh
    x_min, x_max = X_test[:, 0].min() - 1, X_test[:, 0].max() + 1
    y_min, y_max = X_test[:, 1].min() - 1, X_test[:, 1].max() + 1
    xx, yy = np.meshgrid(np.arange(x_min, x_max, h),
                         np.arange(y_min, y_max, h))

    X_hypo = np.column_stack((xx.ravel().astype(np.float32),
                              yy.ravel().astype(np.float32)))
    ret = model.predict(X_hypo)
    if isinstance(ret, tuple):
        zz = ret[1]
    else:
        zz = ret
    zz = zz.reshape(xx.shape)

    plt.contourf(xx, yy, zz, cmap=plt.cm.coolwarm, alpha=0.8)
    plt.scatter(X_test[:, 0], X_test[:, 1], c=y_test, s=200)


plt.figure(figsize=(10, 6))
plot_decision_boundary(model_norm, X, y)
plt.show()

运行结果如下：