latent-space-theories

Runtime error

ludusc commited on May 23, 2023

Commit

50eece1

•

1 Parent(s): 0ed5e86

new threshold for important factor

Files changed (1) hide show

backend/disentangle_concepts.py CHANGED Viewed

@@ -15,7 +15,7 @@ def get_separation_space(type_bin, annotations, df):
     svc = SVC(gamma='auto', kernel='linear')
     svc.fit(x_train, y_train)
     print(svc.score(x_val, y_val))
-    imp_features = (np.abs(svc.coef_) > 2).sum()
     return svc.coef_, imp_features
 def regenerate_images(model, z, decision_boundary, min_epsilon=-3, max_epsilon=3, count=11):

     svc = SVC(gamma='auto', kernel='linear')
     svc.fit(x_train, y_train)
     print(svc.score(x_val, y_val))
+    imp_features = (np.abs(svc.coef_) > 0.01).sum()
     return svc.coef_, imp_features
 def regenerate_images(model, z, decision_boundary, min_epsilon=-3, max_epsilon=3, count=11):