18 lines
380 B
Python
18 lines
380 B
Python
import random
|
|
|
|
class DataSet:
|
|
def __init__(self, func, N=1000) -> None:
|
|
self.train = []
|
|
self.train_answs = []
|
|
self.test = []
|
|
self.test_answs = []
|
|
|
|
for i in range(N//5*4):
|
|
x = random.uniform(1, 9)
|
|
self.train.append(x)
|
|
self.train_answs.append(func(x))
|
|
|
|
for i in range(N//5):
|
|
x = random.uniform(1, 9)
|
|
self.test.append(x)
|
|
self.test_answs.append(func(x)) |