Qmemristor/test_data.py at master · IraitzM/Qmemristor · GitHub

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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
import numpy as np

# define binary digit arrays
zero = [[1,1,1,1],
	   [1,0,0,1],
	   [1,0,0,1],
	   [1,0,0,1],
	   [1,1,1,1]]
zero = np.asarray(zero)

one = [[0,1,1,0],
	   [0,0,1,0],
	   [0,0,1,0],
	   [0,0,1,0],
	   [0,0,1,0]]
one = np.asarray(one)

two = [[0,1,1,0],
	   [0,0,0,1],
	   [0,0,1,0],
	   [0,1,0,0],
	   [1,1,1,1]]
two = np.asarray(two)

three = [[1,1,1,0],
	    [0,0,0,1],
	    [0,1,1,0],
	    [0,0,0,1],
	    [1,1,1,0]]
three = np.asarray(three)

four = [[1,0,1,0],
	   [1,0,1,0],
	   [1,1,1,1],
	   [0,0,1,0],
	   [0,0,1,0]]
four = np.asarray(four)

five = [[0,1,1,1],
	   [0,1,0,0],
	   [0,1,1,1],
	   [0,0,0,1],
	   [1,1,1,1]]
five = np.asarray(five)

six = [[1,1,1,0],
	   [1,0,0,0],
	   [1,1,1,1],
	   [1,0,0,1],
	   [1,1,1,1]]
six = np.asarray(six)

seven = [[1,1,1,1],
	   [0,0,0,1],
	   [0,0,1,0],
	   [0,0,1,0],
	   [0,0,1,0]]
seven = np.asarray(seven)

eight = [[1,1,1,1],
	   [1,0,0,1],
	   [1,1,1,1],
	   [1,0,0,1],
	   [1,1,1,1]]
eight = np.asarray(eight)

nine = [[1,1,1,1],
	   [1,0,0,1],
	   [1,1,1,1],
	   [0,0,0,1],
	   [0,0,0,1]]
nine = np.asarray(nine)


# generate data and labels
data = np.asarray([zero, one, two, three, four, five, six, seven, eight, nine])
labels = [0,1,2,3,4,5,6,7,8,9]