markus/lib-paveit-demo
1
0
Fork 0
Code Pull Requests Activity

remove print

Browse Source
This commit is contained in:
Markus Clauß
2023-06-28 09:03:13 +02:00
parent 04f5845ef1
commit 423ad74d23
1 changed files with 22 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/paveit/postprocessing/citt.py
View File

@@ -54,17 +54,12 @@ def citt(task_id: str):
line_mod = lm.models.LinearModel() line_mod = lm.models.LinearModel()
out = line_mod.fit(data.stiffness, x=data.phase) out = line_mod.fit(data.stiffness, x=data.phase)
print(out.best_values)
Emax = line_mod.eval(out.params, x=0.0) Emax = line_mod.eval(out.params, x=0.0)
Emin = 0 Emin = 0
assert Emin < Emax assert Emin < Emax
print(data.head())
# Fit data # Fit data
mod = lm.models.Model(stiffness_tp26, independent_vars=['f','T']) mod = lm.models.Model(stiffness_tp26, independent_vars=['f','T'])
@@ -105,6 +100,8 @@ def citt(task_id: str):
results = [] results = []
r2 = [] r2 = []
try:
methods = ['leastsq', 'powell'] methods = ['leastsq', 'powell']
for method in methods: for method in methods:
@@ -118,11 +115,16 @@ def citt(task_id: str):
best = np.nanargmax(r2) best = np.nanargmax(r2)
res = results[best].best_values res = results[best].best_values
res['nsamples'] = len(data)
res['task_id'] = task_id
res['stat_r2'] = r2[best] res['stat_r2'] = r2[best]
except:
print('error regression, send default values')
res = mod.valuesdict()
#add metadata
res['nsamples'] = len(data)
res['task_id'] = task_id
res['date'] = datetime.datetime.now() res['date'] = datetime.datetime.now()
print(res) print(res)