Skip to content

Package that returns a company embedding given a company name

License

Notifications You must be signed in to change notification settings

eddiepease/company2vec

Repository files navigation

Company2Vec

CircleCI

This repository generates a company vector embedding given a company name. Many machine learning use cases in business require a numerical representation of company. This repo is designed to aid this process by being easy to understand and generalizable to different companies/industries.

Getting started

To try out the package, follow the steps below:

  • Clone this repository to local machine
  • cd in folder root
  • chmod +x setup.sh to make bash file executable
  • ./setup.sh to run executable - this installs a virtualenv and downloads relevant data
  • Setup a Bing API on Azure and replace the subscription key in config.py (there is a free version)
  • From the command line / terminal, type in:>> python quick_start.py --company_name "<company_name_of_your_choice>" to get the embedding.
  • For short hand notation type in:>> python quick_start.py -c "<company_name_of_your_choice>"

Features

Vector embeddings are often used for natural language processing in machine learning. They are used to represent a concept as a vector - this vector can then be used in a machine learning model. The embedding is created though a combination of an Azure API (to find the company website), scrapy (to do a shallow scrape of the company website) and pre-trained GloVe embeddings.

To return a small number of company embeddings, use quick_start.py. To generate company embeddings at scale, build a web app using kleinapp.py and Dockerfile and deploy this docker image to the cloud.

Documentation

Documentation can be found here.

Contributing

Please do contribute to improve the repository. If you have an issue with the current code/documentation, do open an issue here

Licensing

This project is licensed under MIT License.

About

Package that returns a company embedding given a company name

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published