一、构造数据
#根据rating和age评估价格
def wineprice(rating,age):
peak_age=rating-50
price=rating/2
if age>peak_age:
price=price*(5-(age-peak_age))#过了峰值年,价值降低
else:
price=price*(5*(age+1)/peak_age)#临近峰值年,价值增高
if price<0:
price=0
return price
#构造数据集合:输入——rating,age;输出——price
def wineset1():
rows=[]
for i in range(300):
rating=random()*50+50
age=random()*50
price=wineprice(rating,age)
price*=(random()*0.4+0.8)
rows.append({
'input':(rating,age),'result':price})
return rows
#构造数据集合:输入——rating,age,aisle,bottlesize;输出——price
def wineset2():
rows=[]
for i in range(300):
rating=random()*50+50
age=random()*50
aisle=float(randint(