Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/neuml/paperai
📄 🤖 Semantic search and workflows for medical/scientific papers
https://github.com/neuml/paperai
ai artificial-intelligence document-search machine-learning medical nlp python scientific-papers search txtai
Last synced: 3 days ago
JSON representation
📄 🤖 Semantic search and workflows for medical/scientific papers
- Host: GitHub
- URL: https://github.com/neuml/paperai
- Owner: neuml
- License: apache-2.0
- Created: 2020-07-21T18:33:30.000Z (over 4 years ago)
- Default Branch: master
- Last Pushed: 2024-12-28T20:59:36.000Z (24 days ago)
- Last Synced: 2025-01-11T04:07:06.599Z (10 days ago)
- Topics: ai, artificial-intelligence, document-search, machine-learning, medical, nlp, python, scientific-papers, search, txtai
- Language: Python
- Homepage:
- Size: 1.72 MB
- Stars: 1,359
- Watchers: 26
- Forks: 102
- Open Issues: 2
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
Semantic search and workflows for medical/scientific papers
-------------------------------------------------------------------------------------------------------------------------------------------------------
paperai is a semantic search and workflow application for medical/scientific papers.
Applications range from semantic search indexes that find matches for medical/scientific queries to full-fledged reporting applications powered by machine learning.
paperai and/or NeuML has been recognized in the following articles:
- [Machine-Learning Experts Delve Into 47,000 Papers on Coronavirus Family](https://www.wsj.com/articles/machine-learning-experts-delve-into-47-000-papers-on-coronavirus-family-11586338201)
- [Data scientists assist medical researchers in the fight against COVID-19](https://cloud.google.com/blog/products/ai-machine-learning/how-kaggle-data-scientists-help-with-coronavirus)
- [CORD-19 Kaggle Challenge Awards](https://www.kaggle.com/allen-institute-for-ai/CORD-19-research-challenge/discussion/161447)
## Installation
The easiest way to install is via pip and PyPI
```
pip install paperai
```
Python 3.9+ is supported. Using a Python [virtual environment](https://docs.python.org/3/library/venv.html) is recommended.
paperai can also be installed directly from GitHub to access the latest, unreleased features.
```
pip install git+https://github.com/neuml/paperai
```
See [this link](https://neuml.github.io/txtai/install/#environment-specific-prerequisites) to help resolve environment-specific install issues.
### Docker
Run the steps below to build a docker image with paperai and all dependencies.
```
wget https://raw.githubusercontent.com/neuml/paperai/master/docker/Dockerfile
docker build -t paperai .
docker run --name paperai --rm -it paperai
```
paperetl can be added in to have a single image to index and query content. Follow the instructions to build a [paperetl docker image](https://github.com/neuml/paperetl#docker) and then run the following.
```
docker build -t paperai --build-arg BASE_IMAGE=paperetl --build-arg START=/scripts/start.sh .
docker run --name paperai --rm -it paperai
```
## Examples
The following notebooks and applications demonstrate the capabilities provided by paperai.
### Notebooks
| Notebook | Description | |
|:----------|:-------------|------:|
| [Introducing paperai](https://github.com/neuml/paperai/blob/master/examples/01_Introducing_paperai.ipynb) | Overview of the functionality provided by paperai | [](https://colab.research.google.com/github/neuml/paperai/blob/master/examples/01_Introducing_paperai.ipynb) |
### Applications
| Application | Description |
|:----------|:-------------|
| [Search](https://github.com/neuml/paperai/blob/master/examples/search.py) | Search a paperai index. Set query parameters, execute searches and display results. |
## Building a model
paperai indexes databases previously built with [paperetl](https://github.com/neuml/paperetl). The following shows how to create a new paperai index.
1. (Optional) Create an index.yml file
paperai uses the default txtai embeddings configuration when not specified. Alternatively, an index.yml file can be specified that takes all the same options as a txtai embeddings instance. See the [txtai documentation](https://neuml.github.io/txtai/embeddings/configuration) for more on the possible options. A simple example is shown below.
```
path: sentence-transformers/all-MiniLM-L6-v2
content: True
```
2. Build embeddings index
```
python -m paperai.index
```
The paperai.index process requires an input data path and optionally takes index configuration. This configuration can either be a vector model path or an index.yml configuration file.
## Running queries
The fastest way to run queries is to start a paperai shell
```
paperai
```
A prompt will come up. Queries can be typed directly into the console.
## Building a report file
Reports support generating output in multiple formats. An example report call:
```
python -m paperai.report report.yml 50 md
```
The following report formats are supported:
- Markdown (Default) - Renders a Markdown report. Columns and answers are extracted from articles with the results stored in a Markdown file.
- CSV - Renders a CSV report. Columns and answers are extracted from articles with the results stored in a CSV file.
- Annotation - Columns and answers are extracted from articles with the results annotated over the original PDF files. Requires passing in a path with the original PDF files.
In the example above, a file named report.md will be created. Example report configuration files can be found [here](https://github.com/neuml/cord19q/tree/master/tasks).
## Tech Overview
paperai is a combination of a [txtai](https://github.com/neuml/txtai) embeddings index and a SQLite database with the articles. Each article is parsed into sentences and stored in SQLite along with the article metadata. Embeddings are built over the full corpus.
Multiple entry points exist to interact with the model.
- paperai.report - Builds a report for a series of queries. For each query, the top scoring articles are shown along with matches from those articles. There is also a highlights section showing the most relevant results.
- paperai.query - Runs a single query from the terminal
- paperai.shell - Allows running multiple queries from the terminal