-
Notifications
You must be signed in to change notification settings - Fork 270
/
test_templates.py
48 lines (38 loc) · 1.57 KB
/
test_templates.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
def test_templates():
print("\n >>> start templates ...")
import talos
x, y = talos.templates.datasets.titanic()
x = x[:50]
y = y[:50]
model = talos.templates.models.titanic
p = talos.templates.params.titanic()
talos.Scan(x, y, p, model, 'test_', round_limit=2)
x, y = talos.templates.datasets.iris()
x = x[:50]
y = y[:50]
model = talos.templates.models.iris
p = talos.templates.params.iris()
talos.Scan(x, y, p, model, 'test_', round_limit=2)
x, y = talos.templates.datasets.cervical_cancer()
x = x[:50]
y = y[:50]
model = talos.templates.models.cervical_cancer
p = talos.templates.params.cervical_cancer()
talos.Scan(x, y, p, model, 'test_', round_limit=2)
x, y = talos.templates.datasets.breast_cancer()
x = x[:50]
y = y[:50]
model = talos.templates.models.breast_cancer
p = talos.templates.params.breast_cancer()
talos.Scan(x, y, p, model, 'test_', round_limit=2)
x, y = talos.templates.datasets.icu_mortality(50)
x, y = talos.templates.datasets.telco_churn(.3)
x, y, x1, y1 = talos.templates.datasets.mnist()
x, y = talos.templates.datasets.breast_cancer()
x, y = talos.templates.datasets.cervical_cancer()
x, y = talos.templates.datasets.titanic()
talos.templates.pipelines.breast_cancer(random_method='uniform_mersenne')
talos.templates.pipelines.cervical_cancer(random_method='sobol')
talos.templates.pipelines.iris(random_method='uniform_crypto')
talos.templates.pipelines.titanic(random_method='korobov_matrix')
print("finish templates \n")