diff options
| author | Birte Kristina Friesel <birte.friesel@uos.de> | 2024-01-16 18:18:36 +0100 |
|---|---|---|
| committer | Birte Kristina Friesel <birte.friesel@uos.de> | 2024-01-16 18:18:36 +0100 |
| commit | 14b59f86359d83af700e8d2e91168bb83a148c27 (patch) | |
| tree | 226dc7f28c3b2d1701dec3c50a7d5e70b5af6953 | |
| parent | 7c1f72e3cd100025247958989f55b97f556986a4 (diff) | |
build_dtree: add another assertion for good measure
| -rw-r--r-- | lib/parameters.py | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/lib/parameters.py b/lib/parameters.py index 58af96f..a32eb9f 100644 --- a/lib/parameters.py +++ b/lib/parameters.py @@ -1303,10 +1303,8 @@ class ModelAttribute: else: children.extend((np.array(child_data) - np.mean(child_data)) ** 2) - if np.any(np.isnan(children)): - loss.append(np.inf) - else: - loss.append(np.sum(children)) + assert not np.any(np.isnan(children)) + loss.append(np.sum(children)) if np.all(np.isinf(loss)): if ffs_feasible: |