Skip to main content

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)

Uploaded Source

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

Hashes for tfseqestimator-2.2.1.tar.gz
Algorithm Hash digest
SHA256 85cc955bfad4ec8aea3d3a02d2fa72eea3b5020a7c6afb2e6081f383fc16f410
MD5 346641fea37f17fdf214c95ae504e4ee
BLAKE2b-256 5887723780993a60e7c196de49fc85a1ea73fdf8d7d99bb69197a020b1226110

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page