30 lines
633 B
Python
Executable File
30 lines
633 B
Python
Executable File
import numpy as np
|
|
import random
|
|
from Distribution import Distribution
|
|
class ProbabilitySet:
|
|
def __init__(self,dset,p):
|
|
self.dset = dset
|
|
self.probiablity = p
|
|
|
|
def dataset(self):
|
|
dl = Distribution([1,0],[self.probiablity,])
|
|
v = dl.generator(len(self.dset))
|
|
m = []
|
|
for i in range(len(self.dset)):
|
|
if v[i] == 1:
|
|
m.append(self.dset)
|
|
return m
|
|
|
|
if __name__ == '__main__':
|
|
from timecost import TimeCost
|
|
dset = range(10000000)
|
|
pp = [0.1,0.22,0.334,0.88,0.93]
|
|
tc = TimeCost()
|
|
for p in pp:
|
|
tc.begin('dataset() %f' % p)
|
|
ps = ProbablitySet(dset,p)
|
|
d = ps.dataset()
|
|
tc.end('dataset() %f' % p)
|
|
|
|
tc.show()
|