Skip to main content

seq2class is developed for text classification using LSTM

Project description

seq2class |Build Status| |PyPI version|

seq2class is a one stop solution for text classification. Text classification is made easy via seq2class package using sequential model(LSTM). This package is build by harnesing the capabilities of LSTM(Long short term Memory) model.

Requirement

seq2class support Python 3.6 or newer.

Installation

.. code:: shell

pip install seq2class

Example

This package is being developed for text classification.

data = 'movies.csv'
labels = 'title'
text = 'genres'

s = Sequence2class()
X_train, X_test, y_train, y_test = s.train_test_split(data, labels, text)
trained_model = s.fit_train(X_train, y_train, 500, 50, 7789, 5, 4)
prediction = s.predict(trained_model, X_test, y_train, 4)

where movies.csv is a training file containing text and labels.

References

DataSet information


[1]  movies.csv dataset have been used for research purpose from this `*link* <http://files.grouplens.org/datasets/movielens/ml-latest-small.zip>`.

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

seq2class-0.2.0.tar.gz (1.6 kB view hashes)

Uploaded source

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Huawei Huawei PSF Sponsor Microsoft Microsoft PSF Sponsor NVIDIA NVIDIA PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page