| 1234567891011121314151617181920212223 |
- import random
- def generate_data(num):
- results=[]
- count=0
- while True:
- x1=random.uniform(0,10)
- x2=random.uniform(0,10)
- if abs(x1-x2)<2:
- continue
-
- y=1 if x1>x2 else 0
- results.append(str([[x1,x2],[y]]))
- count+=1
- if count>=num:
- break
- return results
- train_data=generate_data(500)
- test_data=generate_data(100)
- with open("train_data","w") as f :
- f.writelines("\n".join(train_data))
- with open("test_data","w") as f :
- f.writelines("\n".join(test_data))
|
