Sequence estimators for Tensorflow
Project description
tfseqestimator
Sequence estimators for TensorFlow.
Available estimators
- FullSequenceClassifier: one class for a whole sequence
- FullSequenceRegressor: one value for a whole sequence
- SequenceItemsClassifier: one class for each sequence item
- SequenceItemsRegressor: one value for each sequence item
Usage
from tfseqestimator import FullSequenceClassifier, RnnType
import tensorflow.contrib.feature_column as contrib_features
token_sequence = contrib_features.sequence_categorical_column_with_hash_bucket(...)
token_emb = contrib_features.embedding_column(categorical_column=token_sequence, ...)
estimator = FullSequenceClassifier(
sequence_feature_columns=[token_emb],
rnn_type=RnnType.REGULAR_STACKED_LSTM,
rnn_layers=[32, 16]
)
# Input builders
def input_fn_train: # returns x, y
pass
estimator.train(input_fn=input_fn_train, steps=100)
def input_fn_eval: # returns x, y
pass
metrics = estimator.evaluate(input_fn=input_fn_eval, steps=10)
def input_fn_predict: # returns x, None
pass
predictions = estimator.predict(input_fn=input_fn_predict)
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
tfseqestimator-2.2.1.tar.gz
(31.1 kB
view details)
File details
Details for the file tfseqestimator-2.2.1.tar.gz
.
File metadata
- Download URL: tfseqestimator-2.2.1.tar.gz
- Upload date:
- Size: 31.1 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/1.12.1 pkginfo/1.4.2 requests/2.19.1 setuptools/39.1.0 requests-toolbelt/0.8.0 tqdm/4.23.4 CPython/3.6.5
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 |
85cc955bfad4ec8aea3d3a02d2fa72eea3b5020a7c6afb2e6081f383fc16f410
|
|
MD5 |
346641fea37f17fdf214c95ae504e4ee
|
|
BLAKE2b-256 |
5887723780993a60e7c196de49fc85a1ea73fdf8d7d99bb69197a020b1226110
|