diff options
Diffstat (limited to 'lib')
| -rwxr-xr-x | lib/dfatool.py | 22 |
1 files changed, 16 insertions, 6 deletions
diff --git a/lib/dfatool.py b/lib/dfatool.py index 5cece86..41d5d37 100755 --- a/lib/dfatool.py +++ b/lib/dfatool.py @@ -371,16 +371,26 @@ class EnergyModel: if elem['isa'] == 'state': predicted_data = np.array(list(map(lambda x: model_function(name, 'power'), elem['power']))) measures = regression_measures(predicted_data, elem['power']) - print(' power: {:.2f}% / {:.0f} µW'.format( - measures['smape'], measures['mae'] - )) + if 'smape' in measures: + print(' power: {:.2f}% / {:.0f} µW'.format( + measures['smape'], measures['mae'] + )) + else: + print(' power: {:.0f} µW'.format( + measures['mae'] + )) else: for key in ['duration', 'energy', 'rel_energy_prev', 'rel_energy_next']: predicted_data = np.array(list(map(lambda x: model_function(name, key), elem[key]))) measures = regression_measures(predicted_data, elem[key]) - print(' {:10s}: {:.2f}% / {:.0f}'.format( - key, measures['smape'], measures['mae'] - )) + if 'smape' in measures: + print(' {:10s}: {:.2f}% / {:.0f}'.format( + key, measures['smape'], measures['mae'] + )) + else: + print(' {:10s}: {:.0f}'.format( + key, measures['mae'] + )) |