/
print_fit.py
60 lines (48 loc) · 1.66 KB
/
print_fit.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
49
50
51
52
53
54
55
56
57
58
59
60
"Implements functions for raw fit printing from params"
from numpy import exp
# pylint: disable=invalid-name
def print_isma(A, B, alpha, d, K):
"""prints ISMA fit from params"""
print("ISMA fit from params")
string_list = [None]*K
print_string = "1 = "
for k in range(K):
if k > 0:
print(print_string)
print_string = " + "
print_string += "({0:.6g}/w**{1:.6g})".format(exp(alpha[k]*B[k]), alpha[k])
for i in range(d):
print_string += " * (u_{0:d})**{1:.6g}".format(
i + 1, alpha[k]*A[d*k + i]
)
string_list[k] = print_string
print(print_string)
return string_list
# pylint: disable=invalid-name
def print_sma(A, B, alpha, d, K):
"""prints SMA fit from params"""
print("SMA fit from params")
string_list = [None]*K
print_string = "w**{0:.6g} = ".format(alpha)
for k in range(K):
if k > 0:
print(print_string)
print_string = " + "
print_string += "{0:.6g}".format(exp(alpha*B[k]))
for i in range(d):
print_string += " * (u_{0:d})**{1:.6g}".format(i + 1, alpha*A[d*k + i])
string_list[k] = print_string
print(print_string)
return string_list
# pylint: disable=invalid-name
def print_ma(A, B, d, K):
"""prints MA fit from params"""
print("MA fit from params")
string_list = [None]*K
for k in range(K):
print_string = "w = {0:.6g}".format(exp(B[k]))
for i in range(d):
print_string += " * (u_{0:d})**{1:.6g}".format(i + 1, A[d*k + i])
string_list[k] = print_string
print(print_string)
return string_list