Skip to content

TheDying0fLight/boolean-clip

 
 

Repository files navigation

Joint Representations of Natural Language and Complex Queries

This repository contains a CLIP/SigLIP inspired model for comparing data of two different text modalities (here boolean language queries and natural language queries but adaptable to other types of text). There also is a visualization tool for visualizing the embedding space through Umap.

Quickstart

To start a local server for hosting the embedding visualization website run python app.py

To only calculate the embeddings and cache them use python app.py --precalculate

Overview

This repository consists of a CLIP inspired model, which is used to compare Natural Language Queries with Boolean Language Queries. It also contains helper functions for preprocessing data from different sources, and training/evaluating different models.

app.py

By running this file a local server which hosts a website showing embeddings of given data in a 2D plot

By adjusting variables in the beginning of this file the input data and used model can be modified.

eval.ipynb

This notebook contains examples for evaluating models with our evaluation functions

plotting.ipynb

This notebook was used to create plots from data saved to W&B while training models

train.ipynb

A notebook for training/fine-tuning models with HuggingFace libraries

Images

Visualization Tool

VisOverview

Some Training Results

image image

About

Research project which creates a joint embedding space for complex queries (used in systematic reviews) and natural language, to help with evaluating translations between these modalities.

Resources

Stars

0 stars

Watchers

0 watching

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages

  • Python 55.5%
  • Jupyter Notebook 44.5%