Spaces:
Runtime error
Runtime error
fix bug
Browse files
pages/1_Textiles_Disentanglement.py
CHANGED
@@ -32,7 +32,7 @@ with open(annotations_file, 'rb') as f:
|
|
32 |
annotations = pickle.load(f)
|
33 |
|
34 |
concept_vectors = pd.read_csv('./data/stored_vectors/scores_colors_hsv.csv')
|
35 |
-
concept_vectors['vector'] = np.array([float(x) for x in concept_vectors['vector'].str.split(', ')])
|
36 |
concept_vectors['score'] = concept_vectors['score'].astype(float)
|
37 |
concept_vectors = concept_vectors.sort_values('score', ascending=False).reset_index()
|
38 |
print(concept_vectors[['vector', 'score']])
|
|
|
32 |
annotations = pickle.load(f)
|
33 |
|
34 |
concept_vectors = pd.read_csv('./data/stored_vectors/scores_colors_hsv.csv')
|
35 |
+
concept_vectors['vector'] = np.array([np.array([float(xx) for xx in x]) for x in concept_vectors['vector'].str.split(', ')])
|
36 |
concept_vectors['score'] = concept_vectors['score'].astype(float)
|
37 |
concept_vectors = concept_vectors.sort_values('score', ascending=False).reset_index()
|
38 |
print(concept_vectors[['vector', 'score']])
|