12345678910111213141516 |
- # -*- encoding:utf-8 -*-
- import numpy as np
- import random
- def generate():
- X=[[random.randint(0,255) for i in range(0,28)] for j in range(0,28)]
- X=[[0 if s <128 else s for s in ss] for ss in X]
- return str([X,10])
- train_num=6000
- test_num=1000
- with open("train_data_10","w") as f :
- train_data=[ generate() for _ in range(0,train_num)]
- f.writelines("\n".join(train_data))
- with open("test_data_10","w") as f :
- train_data=[ generate() for _ in range(0,test_num)]
- f.writelines("\n".join(train_data))
|