ialab-puc / CuratorNet

CuratorNet: Visually-aware Recommendation of Art Images
MIT License
14 stars 1 forks source link

Problema de sets con Training.ipynb #1

Open isipalma opened 3 years ago

isipalma commented 3 years ago

En el archivo Training.ipynb, en la sección de Concatenate embeddings + z-score normalization, la línea de artwork_ids_set.update(embedding['index2id'])se realiza mal el update, ya que queda un set del tipo:

en vez de

Esto provoca un KeyError más adelante, en Load clusters imagen

La solución que me sirvió fue lo siguiente:

imagen

Simplemente agregué .values() para que el set recibiera los artwork_id y no los index.

Saludos!

victoraccete commented 2 years ago

Thank you!