ludusc commited on
Commit
ec4ccb0
1 Parent(s): 708c2f6
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']])