# Hyperbolic Probe

## Requirements

    $ pip install -r requirements.txt

## Sentiment Visualization

Probe checkpoints are in the folder `./SentimentProbe/checkpoint`.

Use the Jupyter Notebook `Sentiment-Visualization-Poincare/Euclidean.ipynb` to visualize the sentiment subspace.

Play with it by changing the input text.

### Example

<p align="left">
  <img src="readme-asset/header.png" width="400" title="hover text" alt="BERT is a good model">
</p>

_BERT is a good model_

## Run Experiment

Use `run.py`, which can be found in the sub-folders `./SyntaxProbe` and `./SentimentProbe`.
