{ "info": { "author": "Ryuya Ikeda", "author_email": "rikeda71@gmail.com", "bugtrack_url": null, "classifiers": [ "License :: OSI Approved :: MIT License", "Programming Language :: Python :: 3.6", "Programming Language :: Python :: 3.7", "Topic :: Text Processing" ], "description": "# Torch CRF\n[![CircleCI](https://circleci.com/gh/s14t284/TorchCRF.svg?style=svg)](https://circleci.com/gh/s14t284/TorchCRF) [![Coverage Status](https://coveralls.io/repos/github/s14t284/TorchCRF/badge.svg)](https://coveralls.io/github/s14t284/TorchCRF)\n\nImplementation of CRF (Conditional Random Fields) in PyTorch 1.0\n\n## Requirements\n\n- python3 (>=3.6)\n- PyTorch 1.0\n\n## Installation\n\n $ pip install TorchCRF\n\n## Usage\n\n```python\n>>> import torch\n>>> from TorchCRF import CRF\n>>> batch_size = 2\n>>> sequence_size = 3\n>>> num_labels = 5\n>>> mask = torch.FloatTensor([[1, 1, 1], [1, 1, 0]]) # (batch_size. sequence_size)\n>>> labels = torch.LongTensor([[0, 2, 3], [1, 4, 1]]) # (batch_size, sequence_size)\n>>> hidden = torch.randn((batch_size, sequence_size, num_labels), requires_grad=True)\n>>> crf = CRF(num_labels)\n```\n\n### Computing log-likelihood (used where forward)\n\n```python\n>>> crf.forward(hidden, labels, mask)\ntensor([-7.6204, -3.6124], grad_fn=)\n```\n\n### Decoding (predict labels of sequences)\n\n```python\n>>> crf.viterbi_decode(hidden, mask)\n[[0, 2, 2], [4, 0]]\n```\n\n## License\n\nMIT\n\n## References\n\n- [threelittlemonkeys/lstm-crf-pytorch](https://github.com/threelittlemonkeys/lstm-crf-pytorch)\n- [kmkurn/pytorch-crf](https://github.com/kmkurn/pytorch-crf)\n\n\n", "description_content_type": "text/markdown", "docs_url": null, "download_url": "", "downloads": { "last_day": -1, "last_month": -1, "last_week": -1 }, "home_page": "https://github.com/s14t284/TorchCRF", "keywords": "crf,conditional random fields,nlp,natural language processing", "license": "MIT", "maintainer": "", "maintainer_email": "", "name": "TorchCRF", "package_url": "https://pypi.org/project/TorchCRF/", "platform": "", "project_url": "https://pypi.org/project/TorchCRF/", "project_urls": { "Homepage": "https://github.com/s14t284/TorchCRF" }, "release_url": "https://pypi.org/project/TorchCRF/1.0.4/", "requires_dist": [ "numpy", "torch (>=1.0.0)" ], "requires_python": "", "summary": "An Implementation of Conditional Random Fields in pytorch", "version": "1.0.4" }, "last_serial": 5598035, "releases": { "1.0.0": [ { "comment_text": "", "digests": { "md5": "4e76928b160e5b9cf0bed29a006e43ff", "sha256": "473cce22162b5c4773b503791781831f2d19bcc2fb6abd0774fb99275565e5e1" }, "downloads": -1, "filename": "TorchCRF-1.0.0-py3-none-any.whl", "has_sig": false, "md5_digest": "4e76928b160e5b9cf0bed29a006e43ff", "packagetype": "bdist_wheel", "python_version": "py3", "requires_python": null, "size": 6229, "upload_time": "2019-03-19T15:58:25", "url": "https://files.pythonhosted.org/packages/60/1a/3f05758c074d7cc616e95111b33f4a74767d47866d65d26e0fcaa205d95d/TorchCRF-1.0.0-py3-none-any.whl" }, { "comment_text": "", "digests": { "md5": "65a046a2d3503df45e6cd150d6a6823f", "sha256": "c6c44766fb55ac118b9d9c4c1e664ed570feb471c7c828cf804363ef05957c9e" }, "downloads": -1, "filename": "TorchCRF-1.0.0.tar.gz", "has_sig": false, "md5_digest": "65a046a2d3503df45e6cd150d6a6823f", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 5458, "upload_time": "2019-03-19T15:58:27", "url": "https://files.pythonhosted.org/packages/be/c6/c710531f98898e191a7aaff8a725f3e54568deb52ec8b21ce85e644af647/TorchCRF-1.0.0.tar.gz" } ], "1.0.1": [ { "comment_text": "", "digests": { "md5": "374f251949121d3ea53d31c6742e3bc0", "sha256": "a0b79c723677e58b1c3586ed003ecf7c59840b5210cf606f3f92eda54c8bd627" }, "downloads": -1, "filename": "TorchCRF-1.0.1-py3-none-any.whl", "has_sig": false, "md5_digest": "374f251949121d3ea53d31c6742e3bc0", "packagetype": "bdist_wheel", "python_version": "py3", "requires_python": null, "size": 5781, "upload_time": "2019-03-19T16:42:07", "url": "https://files.pythonhosted.org/packages/b8/26/e42ee0b2106b85cf7842947915555062bd95c3584a23269650bf12c3a556/TorchCRF-1.0.1-py3-none-any.whl" }, { "comment_text": "", "digests": { "md5": "de5c48973fef3f8691f3629dfed0092b", "sha256": "4c110d15ed1cc33bf714f3a75dc4c21f597a01f657b32b8c96aef190b5ab1575" }, "downloads": -1, "filename": "TorchCRF-1.0.1.tar.gz", "has_sig": false, "md5_digest": "de5c48973fef3f8691f3629dfed0092b", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 5657, "upload_time": "2019-03-19T16:42:08", "url": "https://files.pythonhosted.org/packages/d4/91/7f87f3ddbf4665067eb166f1a606e74a99aad1d6fa9127efce79a1dbfa9a/TorchCRF-1.0.1.tar.gz" } ], "1.0.2": [ { "comment_text": "", "digests": { "md5": "acb5a485a3ae9b8441e8fb36d901dfbd", "sha256": "9550b84cf97aecde9185102746c50d9b234cd932c02608624b524ec87833cd1a" }, "downloads": -1, "filename": "TorchCRF-1.0.2-py3-none-any.whl", "has_sig": false, "md5_digest": "acb5a485a3ae9b8441e8fb36d901dfbd", "packagetype": "bdist_wheel", "python_version": "py3", "requires_python": null, "size": 5767, "upload_time": "2019-04-04T08:10:52", "url": "https://files.pythonhosted.org/packages/b9/2b/d072ca804258eaf8bb0fbf3a4bb3909821422f3d247f889116c81444dc98/TorchCRF-1.0.2-py3-none-any.whl" }, { "comment_text": "", "digests": { "md5": "e1fd558bffd20e0bdb1196be2d3560a1", "sha256": "692fc9e79a5a76e6db5e012e14dd439a7a119e231231744d9f256424f9674235" }, "downloads": -1, "filename": "TorchCRF-1.0.2.tar.gz", "has_sig": false, "md5_digest": "e1fd558bffd20e0bdb1196be2d3560a1", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 5646, "upload_time": "2019-04-04T08:10:53", "url": "https://files.pythonhosted.org/packages/ab/27/a9eb63e22ccd4a8176723230a93c7ff9f57750e0d43f4530fabbaf9d28de/TorchCRF-1.0.2.tar.gz" } ], "1.0.3": [ { "comment_text": "", "digests": { "md5": "2775a8b60b0a757f1fa4dfb65f092e7c", "sha256": "1784cb7cf39f7435baebbba9a12945676352b6ce8a8796cde619bb3109fb0b53" }, "downloads": -1, "filename": "TorchCRF-1.0.3-py3-none-any.whl", "has_sig": false, "md5_digest": "2775a8b60b0a757f1fa4dfb65f092e7c", "packagetype": "bdist_wheel", "python_version": "py3", "requires_python": null, "size": 5856, "upload_time": "2019-07-29T02:57:04", "url": "https://files.pythonhosted.org/packages/58/f9/b9e7673f3ec810890703a4f595873dd802c53b447a36c81b520750bbebfa/TorchCRF-1.0.3-py3-none-any.whl" }, { "comment_text": "", "digests": { "md5": "1050914891f85c88d05c4d299362e1ad", "sha256": "23c5401c58a36d4b816f06c73fabd839f6a75ecb2d2b5b8661b9aeac320d7e95" }, "downloads": -1, "filename": "TorchCRF-1.0.3.tar.gz", "has_sig": false, "md5_digest": "1050914891f85c88d05c4d299362e1ad", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 5743, "upload_time": "2019-07-29T02:57:06", "url": "https://files.pythonhosted.org/packages/6c/11/aae32d57d2d60ab8dca0031c784173b7f1811c340504cc89c764fbf5f3d8/TorchCRF-1.0.3.tar.gz" } ], "1.0.4": [ { "comment_text": "", "digests": { "md5": "619d38c27307012d67a6fa70329ddf59", "sha256": "07d537789211de1ba5a47a7b0504fcf7eccafe0d4f4882ef9cd403c2d6522e97" }, "downloads": -1, "filename": "TorchCRF-1.0.4-py3-none-any.whl", "has_sig": false, "md5_digest": "619d38c27307012d67a6fa70329ddf59", "packagetype": "bdist_wheel", "python_version": "py3", "requires_python": null, "size": 5901, "upload_time": "2019-07-29T07:30:09", "url": "https://files.pythonhosted.org/packages/15/b7/3df56591dd533805a1be5607abb7c1e3255ac2a6d241518246f7a573d32b/TorchCRF-1.0.4-py3-none-any.whl" }, { "comment_text": "", "digests": { "md5": "769292b4fcdfb072e7f6c6d4f7eadec9", "sha256": "1f18362a068d9b38abc740af5bf97437373f9947de41b5fe3412c18e1335f89e" }, "downloads": -1, "filename": "TorchCRF-1.0.4.tar.gz", "has_sig": false, "md5_digest": "769292b4fcdfb072e7f6c6d4f7eadec9", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 5789, "upload_time": "2019-07-29T07:30:10", "url": "https://files.pythonhosted.org/packages/0e/b5/3d5b32fba84513afc5d2fef1d936313fd46a9ba2a5153529fa169b0582a5/TorchCRF-1.0.4.tar.gz" } ] }, "urls": [ { "comment_text": "", "digests": { "md5": "619d38c27307012d67a6fa70329ddf59", "sha256": "07d537789211de1ba5a47a7b0504fcf7eccafe0d4f4882ef9cd403c2d6522e97" }, "downloads": -1, "filename": "TorchCRF-1.0.4-py3-none-any.whl", "has_sig": false, "md5_digest": "619d38c27307012d67a6fa70329ddf59", "packagetype": "bdist_wheel", "python_version": "py3", "requires_python": null, "size": 5901, "upload_time": "2019-07-29T07:30:09", "url": "https://files.pythonhosted.org/packages/15/b7/3df56591dd533805a1be5607abb7c1e3255ac2a6d241518246f7a573d32b/TorchCRF-1.0.4-py3-none-any.whl" }, { "comment_text": "", "digests": { "md5": "769292b4fcdfb072e7f6c6d4f7eadec9", "sha256": "1f18362a068d9b38abc740af5bf97437373f9947de41b5fe3412c18e1335f89e" }, "downloads": -1, "filename": "TorchCRF-1.0.4.tar.gz", "has_sig": false, "md5_digest": "769292b4fcdfb072e7f6c6d4f7eadec9", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 5789, "upload_time": "2019-07-29T07:30:10", "url": "https://files.pythonhosted.org/packages/0e/b5/3d5b32fba84513afc5d2fef1d936313fd46a9ba2a5153529fa169b0582a5/TorchCRF-1.0.4.tar.gz" } ] }