File: bench_domain.py

package info (click to toggle)
orange3 3.40.0-2
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 15,912 kB
  • sloc: python: 162,745; ansic: 622; makefile: 322; sh: 93; cpp: 77
file content (44 lines) | stat: -rw-r--r-- 1,517 bytes parent folder | download | duplicates (3)
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
import numpy as np

from Orange.data import DomainConversion, Domain, Table, \
    ContinuousVariable, DiscreteVariable
from Orange.preprocess import Discretize, EqualFreq, Normalize

from .base import Benchmark, benchmark


class BenchConversion(Benchmark):

    def setUp(self):
        cols = 1000
        rows = 100
        cont = [ContinuousVariable(str(i)) for i in range(cols)]
        disc = [DiscreteVariable("D" + str(i), values=("1", "2")) for i in range(cols)]
        self.domain = Domain(cont + disc)
        self.domain_x = Domain(list(self.domain.attributes) + [ContinuousVariable("x")])
        self.single = Domain([ContinuousVariable("0")])
        self.table = Table.from_numpy(
            self.domain,
            np.random.RandomState(0).randint(0, 2, (rows, len(self.domain.variables))))
        self.discretized_domain = Discretize(EqualFreq(n=3))(self.table).domain
        self.normalized_domain = Normalize()(self.table).domain

    @benchmark(number=5)
    def bench_full(self):
        DomainConversion(self.domain, self.domain_x)

    @benchmark(number=5)
    def bench_selection(self):
        DomainConversion(self.domain, self.single)

    @benchmark(number=5)
    def bench_transform_copy(self):
        self.table.transform(self.domain_x)

    @benchmark(number=5)
    def bench_transform_discretize(self):
        self.table.transform(self.discretized_domain)

    @benchmark(number=5)
    def bench_transform_normalize(self):
        self.table.transform(self.normalized_domain)