Spaces:
Runtime error
Runtime error
revert to orig
Browse files
backend/disentangle_concepts.py
CHANGED
@@ -35,7 +35,7 @@ def get_separation_space(type_bin, annotations, df, samples=100, method='LR', C=
|
|
35 |
print('Val performance SVM', svc.score(x_val, y_val))
|
36 |
imp_features = (np.abs(svc.coef_) > 0.2).sum()
|
37 |
imp_nodes = np.where(np.abs(svc.coef_) > 0.2)[1]
|
38 |
-
return svc.coef_, imp_features, imp_nodes, np.round(clf.score(x_val, y_val),2)
|
39 |
elif method == 'LR':
|
40 |
clf = LogisticRegression(random_state=0, C=C)
|
41 |
clf.fit(x_train, y_train)
|
@@ -145,6 +145,6 @@ def get_concepts_vectors(concepts, annotations, df, samples=100, method='LR', C=
|
|
145 |
# )
|
146 |
|
147 |
# projection = reducer.fit_transform(vectors)
|
148 |
-
nodes_in_common =
|
149 |
return vectors, nodes_in_common, performances
|
150 |
|
|
|
35 |
print('Val performance SVM', svc.score(x_val, y_val))
|
36 |
imp_features = (np.abs(svc.coef_) > 0.2).sum()
|
37 |
imp_nodes = np.where(np.abs(svc.coef_) > 0.2)[1]
|
38 |
+
return svc.coef_ / np.linalg.norm(clf.coef_), imp_features, imp_nodes, np.round(clf.score(x_val, y_val),2)
|
39 |
elif method == 'LR':
|
40 |
clf = LogisticRegression(random_state=0, C=C)
|
41 |
clf.fit(x_train, y_train)
|
|
|
145 |
# )
|
146 |
|
147 |
# projection = reducer.fit_transform(vectors)
|
148 |
+
nodes_in_common = set.intersection(*important_nodes)
|
149 |
return vectors, nodes_in_common, performances
|
150 |
|