optimal-opinion-python-class/test_class_Nagents.py at master · florian-die/optimal-opinion-python-class · 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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
import numpy as np
import opinion_class as op

Y0 = np.array([0.0,3,3.5,4.0,12.0,13.0,14.0])

my_op = op.opinion(Y0)

my_op.kh = 0.0
my_op.mu = 1.0
my_op.sat = np.infty
my_op.eta = 0.5

my_op.rootmethod = 'lm'

#my_op.atol = 1e-14
#my_op.rtol = 1e-10

P0 = np.zeros((my_op.N,))
P0[0] = -0.491088
P0[-1] = 1.91331

tf0 = np.array([16.67589717602594845])

print("Initial educated guess")
my_op.trace(P0,tf0)

print("Refining around guess")
sol0, P1, tf1 = my_op.solve(P0,tf0,trace=True,echo=True)

#%%
''' interactions '''
P2 = P1.copy()
P2[1:-1] = 0.01
tf2 = tf1.copy()

my_op.mu = 1.0
my_op.kh = 0.0
my_op.sat = np.infty

my_op.rootmethod = 'lm'
#my_op.rootmethod = 'hybr'

kh_end = 1.0
kh_step = 0.1

while my_op.kh < kh_end:

    my_op.kh += kh_step
    if my_op.kh > kh_end:
        my_op.kh = kh_end
    print(my_op.kh)

    sol1, P2, tf2 = my_op.solve(P2,tf2,trace=False,echo=False)
    print(sol1.success)

print("Solution with interactions")
my_op.trace(P2,tf2)

#%%
''' control '''
P3 = P2.copy()
tf3 = tf2.copy()

#my_op.rootmethod = 'hybr'
my_op.rootmethod = 'lm'

my_op.sat = 2.0
my_op.mu = 1.0
my_op.kh = 1.0

mu_end = 0.1
mu_step = 0.1

while my_op.mu > mu_end:

    my_op.mu -= mu_step
    if my_op.mu < mu_end:
        my_op.mu = mu_end
    print(my_op.mu)

    sol2, P3, tf3 = my_op.solve(P3,tf3,trace=False,echo=False)
    print(sol2.success)

print("Solution without control cost")
my_op.trace(P3,tf3)

#%%
P4 = P3.copy()
tf4 = tf3.copy()

mu_end = 0.01
mu_step = 0.01

while my_op.mu > mu_end:

    my_op.mu -= mu_step
    if my_op.mu < mu_end:
        my_op.mu = mu_end
    print(my_op.mu)

    sol3, P4, tf4 = my_op.solve(P4,tf4,trace=False,echo=False)
    print(sol2.success)

print("Solution without control cost")
my_op.trace(P4,tf4)