Skip to content

Metadata embeddings

Latest
Latest
Compare
Choose a tag to compare
@nikhilwoodruff nikhilwoodruff released this 02 Apr 13:43
· 1886 commits to master since this release
embeddings
7121e66

This release contains the OpenAI GPT parameter embeddings for parameters and variables from PolicyEngine UK and PolicyEngine US. The model used to embed the parameter metadata was sentence-transformers/all-MiniLM-L6-v2. This release is in two parts:

  • metadata.csv.gz: one row per unique object, columns for name, country_id and type.
  • embeddings.h5: an array of size (number of objects, embedding dimension)
Assets 5
Loading