生成随机产生的训练数据

最新推荐文章于 2024-12-21 09:43:31 发布

原创

最新推荐文章于 2024-12-21 09:43:31 发布 · 1.3k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#机器学习

该博客介绍如何用Python创建包含三个类别的随机训练数据集。通过numpy和matplotlib库生成不同分布的数据点，并根据特定条件分配不同的类别标签，最后将数据写入文件并绘制散点图进行可视化。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

生成随机产生的训练数据#createDist.py

from numpy import *

import matplotlib
import matplotlib.pyplot as plt
from matplotlib.patches import Rectangle


n = 1000 #number of points to create
xcord = zeros((n))#create a shuzu,length=n
ycord = zeros((n))
markers =[]#used to give different date with different size
colors =[]#used to give different date with different color
fw = open('testSet.txt','w')#open a file in a write way
for i in range(n):#give n=1000 date diferent value
    [r0,r1] = random.standard_normal(2)#create a list whitch has two elements
    myClass = random.uniform(0,1)#used to classify
    if (myClass <= 0.16):
        fFlyer = random.uniform(22000, 60000)
        tats = 3 + 1.6*r1
        markers.append(20)
        colors.append(2.1)
        classLabel = 1 #'didntLike'
        print (("%d, %f, class1"