Skip to content

Latest commit

 

History

History
21 lines (13 loc) · 605 Bytes

README.md

File metadata and controls

21 lines (13 loc) · 605 Bytes

New-Software-Identification

To install run:

python setup.py install

You also need to install pytorch, go here for instructions or if you have anaconda run:

conda install pytorch torchvision -c soumith

I also highly recommend installing Cuda if your machine has a GPU

Planned Features

  • Make the number of columns variable
  • Allow online W2V updating
  • Make training easily resumable
  • Add attention to the model
  • Improve example code and analysis notebook
  • Allow more control of print statements e.g. --verbose