torch utils from stylegan3

backend/disentangle_concepts.py

@@ -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 = len(np.abs(svc.coef_) > 2)
+    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):

requirements.txt

@@ -13,5 +13,7 @@ tqdm==4.64.1
 transformers==4.25.1
 scikit-learn
 altair==4.0
-torch-utils
-opencv-python
+#torch-utils
+opencv-python
+stylegan3.dnnlib
+stylegan3.torch_utils