lowered threshold

backend/disentangle_concepts.py

@@ -25,8 +25,8 @@ def get_separation_space(type_bin, annotations, df, samples=100, method='LR', C=
         clf = LogisticRegression(random_state=0, C=C)
         clf.fit(x_train, y_train)
         print('Val performance logistic regression', clf.score(x_val, y_val))
-        imp_features = (np.abs(clf.coef_) > 0.2).sum()
-        imp_nodes = np.where(np.abs(clf.coef_) > 0.2)[1]
+        imp_features = (np.abs(clf.coef_) > 0.15).sum()
+        imp_nodes = np.where(np.abs(clf.coef_) > 0.15)[1]
         return clf.coef_ / np.linalg.norm(clf.coef_), imp_features, imp_nodes