Commit 9d3df708 authored by Valentin Pelloin's avatar Valentin Pelloin
Browse files

added requirements.txt & shifted pmi now with k=1 (no shifting, better results)

parent 674e543b
......@@ -3,6 +3,9 @@ from setuptools import setup
with open("README.md", "r") as fh:
long_description = fh.read()
with open("requirements.txt", "r") as f:
required = f.read().splitlines()
setup(name='svd2vec',
version='0.2',
description='A library that converts words to vectors using PMI and SVD',
......@@ -17,5 +20,6 @@ setup(name='svd2vec',
classifiers=[
"Programming Language :: Python :: 3",
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",]
"Operating System :: OS Independent"],
install_requires=required
)
......@@ -79,7 +79,7 @@ class svd2vec:
window=10,
dyn_window_weight=WINDOW_WEIGHT_WORD2VEC,
cds_alpha=0.75,
neg_k_shift=5,
neg_k_shift=1,
eig_p_weight=0,
nrm_type=NRM_SCHEME_ROW,
sub_threshold=1e-5,
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment