两类正太分布模式的贝叶斯分类

街头~神秘人

于 2022-12-29 20:38:05 发布

阅读量377

点赞数

分类专栏：模式识别与机器学习文章标签：分类 python

本文链接：https://blog.csdn.net/weixin_44820058/article/details/128487643

版权

模式识别与机器学习专栏收录该内容

2 篇文章 0 订阅

订阅专栏

在这里插入图片描述

import numpy as np
import matplotlib.pyplot as plt
a = np.array([[0.,2.,2.,0.], [0.,0.,2.,2.]], dtype=np.float64)
b = np.array([[4.,6.,6.,4.], [4.,4.,6.,6.]], dtype=np.float64)

a_t=np.matrix(a)
b_t = np.matrix(b)
m1 = np.matrix(a.mean(axis=1)).T
m2 = np.matrix(b.mean(axis=1)).T

c1 = np.cov(a_t) / 4 * 3
c2 = np.cov(b_t) / 4 * 3
c1_i = np.linalg.inv(c1)
c2_i = np.linalg.inv(c2)
c_i =c1_i

d1 = np.matmul((m1-m2).T, c_i)
k1 = 1/2 * np.matmul(np.matmul(m1.T, c_i),m1) - 1/2 * np.matmul(np.matmul(m2.T, c_i), m2)

x = np.arange(0,7,1)
y = k1[0,0]/d1[0,1] - (d1[0,0]*x)/d1[0,1]

plt.plot(a[0],a[1],"bo")
plt.plot(b[0], b[1], "go")
plt.plot(x,y)
plt.show()