Excel Integration with SpaCy. Includes, Entity training, Entity matcher pipe.
Project description
ExcelCy
ExcelCy is a NER trainer from XLSX, PDF, DOCX, PPT, PNG or JPG. ExcelCy uses spaCy framework to match Entity with PhraseMatcher or Matcher in regular expression.
ExcelCy is convenience
This is example taken from spaCy documentation, Simple Style Training. It demonstrates how to train NER using spaCy:
import spacy
import random
TRAIN_DATA = [
("Uber blew through $1 million a week", {'entities': [(0, 4, 'ORG')]}), # note: it is required to supply the character position
("Google rebrands its business apps", {'entities': [(0, 6, "ORG")]})] # note: it is required to supply the character position
nlp = spacy.blank('en')
optimizer = nlp.begin_training()
for i in range(20):
random.shuffle(TRAIN_DATA)
for text, annotations in TRAIN_DATA:
nlp.update([text], [annotations], sgd=optimizer)
nlp.to_disk('test_model')
The TRAIN_DATA, describes sentences and annotated entities to be trained. It is cumbersome to always count the characters. With ExcelCy, (start,end) characters can be omitted.
# install excelcy
# pip install excelcy
# download the en model from spacy
# python -m spacy download en"
# run this inside python or file
from excelcy import ExcelCy
# Test: John is the CEO of this_is_a_unique_company_name
excelcy = ExcelCy()
# by default it is assume the nlp_base using model en_core_web_sm
# excelcy.storage.config = Config(nlp_base='en_core_web_sm')
doc = excelcy.nlp('John is the CEO of this_is_a_unique_company_name')
# it will show no company entities
print([(ent.label_, ent.text) for ent in doc.ents])
# run this in root of repo or https://github.com/kororo/excelcy/raw/master/tests/data/test_data_01.xlsx
excelcy = ExcelCy.execute(file_path='tests/data/test_data_01.xlsx')
# use the nlp object as per spaCy API
doc = excelcy.nlp('John is the CEO of this_is_a_unique_company_name')
# now it recognise the company name
print([(ent.label_, ent.text) for ent in doc.ents])
# NOTE: if not showing, remember, it may be required to increase the "train_iteration" or lower the "train_drop", the "config" sheet in Excel
ExcelCy is friendly
By default, ExcelCy training is divided into phases, the example Excel file can be found in tests/data/test_data_01.xlsx:
1. Discovery
The first phase is to collect sentences from data source in sheet "source". The data source can be either:
- Text: Direct sentence values.
- Files: PDF, DOCX, PPT, PNG or JPG will be parsed using textract.
Note: See textract source examples in tests/data/test_data_03.xlsx Note: Dependencies "textract" is not included in the ExcelCy, it is required to add manually
2. Preparation
Next phase, the Gold annotation needs to be defined in sheet "prepare", based on:
- Current Data Model: Using spaCy API of nlp(sentence).ents
- Phrase pattern: Robbie, Uber, Google, Amazon
- Regex pattern: ^([0-1]?[0-9]|2[0-3]):[0-5][0-9]$
All annotations in here are considered as Gold annotations, which described in here.
3. Training
Main phase of NER training, which described in Simple Style Training. The data is iterated from sheet "train", check sheet "config" to control the parameters.
4. Consolidation
The last phase, is to test/save the results and repeat the phases if required.
ExcelCy is flexible
Need more specific export and phases? It is possible to control it using phase API. This is the illustration of the real-world scenario:
-
Train from tests/data/test_data_05.xlsx
# download the dataset $ wget https://github.com/kororo/excelcy/raw/master/tests/data/test_data_05.xlsx # this will create a directory and file "export/train_05.xlsx" $ excelcy execute test_data_05.xlsx
-
Open the result in "export/train_05.xlsx", it shows all identified sentences from source given. However, there is error in the "Himalayas" as identified as "PRODUCT".
-
To fix this, add phrase matcher for "Himalayas = FAC". It is illustrated in tests/data/test_data_05a.xlsx
-
Train again and check the result in "export/train_05a.xlsx"
# download the dataset $ wget https://github.com/kororo/excelcy/raw/master/tests/data/test_data_05a.xlsx # this will create a directory "nlp/data" and file "export/train_05a.xlsx" $ excelcy execute test_data_05a.xlsx
-
Check the result that there is backed up nlp data model in "nlp" and the result is corrected in "export/train_05a.xlsx"
-
Keep training the data model, if there is unexpected behaviour, there is backup data model in case needed.
ExcelCy is comprehensive
Under the hood, ExcelCy has strong and well-defined data storage. At any given phase above, the data can be inspected.
from excelcy import ExcelCy
from excelcy.storage import Config
# Test: John is the CEO of this_is_a_unique_company_name
excelcy = ExcelCy()
excelcy.storage.config = Config(nlp_base='en_core_web_sm', train_iteration=10, train_drop=0.2)
doc = excelcy.nlp('John is the CEO of this_is_a_unique_company_name')
# showing no ORG
print([(ent.label_, ent.text) for ent in doc.ents])
excelcy.storage.source.add(kind='text', value='John is the CEO of this_is_a_unique_company_name')
excelcy.discover()
excelcy.storage.prepare.add(kind='phrase', value='this_is_a_unique_company_name', entity='ORG')
excelcy.prepare()
excelcy.train()
doc = excelcy.nlp('John is the CEO of this_is_a_unique_company_name')
# ORG now is recognised
print([(ent.label_, ent.text) for ent in doc.ents])
# NOTE: if not showing, remember, it may be required to increase the "train_iteration" or lower the "train_drop", the "config" sheet in Excel
Features
- Load multiple data sources such as Word documents, PowerPoint presentations, PDF or images.
- Import/Export configuration with JSON, YML or Excel.
- Add custom Entity labels.
- Rule based phrase matching using PhraseMatcher
- Rule based matching using regex + Matcher
- Train Named Entity Recogniser with ease
Install
Either use the famous pip or clone this repository and execute the setup.py file.
$ pip install excelcy
# ensure you have the language model installed before
$ spacy download en
Train
To train the spaCy model:
from excelcy import ExcelCy
excelcy = ExcelCy.execute(file_path='test_data_01.xlsx')
Note: tests/data/test_data_01.xlsx
CLI
ExelCy has basic CLI command for execute:
$ excelcy execute https://github.com/kororo/excelcy/raw/master/tests/data/test_data_01.xlsx
Test
Run test by installing packages and run tox
$ pip install poetry tox
$ tox
$ tox -e py36 -- tests/test_readme.py
For hot-reload development coding
$ npm i -g nodemon
$ nodemon
Data Definition
ExcelCy has data definition which expressed in api.yml. As long as, data given in this specific format and structure, ExcelCy will able to support any type of data format. Check out, the Excel file format in api.xlsx. Data classes are defined with attrs, check in storage.py for more detail.
Publishing
# this is note for contributors
# ensure locally tests all running
npm run test
# prepare for new version
poetry version 0.4.1
npm run export
# make changes in the git, especially release branch and check in the travis
# https://travis-ci.com/github/kororo/excelcy
# if all goes well, push to master
FAQ
What is that idx columns in the Excel sheet?
The idea is to give reference between two things. Imagine in sheet "train", like to know where the sentence generated from in sheet "source". And also, the nature of Excel, you can sort things, this is the safe guard to keep things in the correct order.
Can ExcelCy import/export to X, Y, Z data format?
ExcelCy has strong and well-defined data storage, thanks to attrs. It is possible to import/export data in any format.
Error: ModuleNotFoundError: No module named 'pip'
There are lots of possibility on this. Try to lower pip version (it was buggy for version 19.0.3).
ExcelCy accepts suggestions/ideas?
Yes! Please submit them into new issue with label "enhancement".
Acknowledgement
This project uses other awesome projects:
- attrs: Python Classes Without Boilerplate.
- pyexcel: Single API for reading, manipulating and writing data in csv, ods, xls, xlsx and xlsm files.
- pyyaml: The next generation YAML parser and emitter for Python.
- spacy: Industrial-strength Natural Language Processing (NLP) with Python and Cython.
- textract: extract text from any document. no muss. no fuss.
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
File details
Details for the file excelcy-0.4.1.tar.gz
.
File metadata
- Download URL: excelcy-0.4.1.tar.gz
- Upload date:
- Size: 24.9 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: poetry/1.0.5 CPython/3.7.7 Darwin/19.6.0
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 7249e11d92eb26f0f5edee36774a884393ff87b54f62cf61fe02302d42933c93 |
|
MD5 | 039ebbd929958940befe7892166de1a2 |
|
BLAKE2b-256 | 7156f82edce277aa5405ff39c12b01145166134e8e65f9479d7eb0deb835d817 |
File details
Details for the file excelcy-0.4.1-py3-none-any.whl
.
File metadata
- Download URL: excelcy-0.4.1-py3-none-any.whl
- Upload date:
- Size: 25.9 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: poetry/1.0.5 CPython/3.7.7 Darwin/19.6.0
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 49730f94a09e3940e3928bf109dcd62e6301b5ca8481c7a6cc97a92f95e0edde |
|
MD5 | 6a6e6e30d3c8d1ecb526a22145aa4f12 |
|
BLAKE2b-256 | 40af610d7d6089e902d18756a710832a5bcba76105c34b9ce7647fb31c48023c |