tensortrade.features.indicators.talib_indicator module

class tensortrade.features.indicators.talib_indicator.TAlibIndicator(indicators, lows=None, highs=None)[source]

Bases: tensortrade.features.feature_transformer.FeatureTransformer

Adds one or more TAlib indicators to a data frame, based on existing open, high, low, and close column values.

transform(X, input_space)[source]

Transform the data set and return a new data frame.

Parameters
  • X (DataFrame) – The set of data to transform.

  • input_space (Space) – A gym.Space matching the shape of the pipeline’s input.

Return type

DataFrame

Returns

A transformed data frame.

transform_space(input_space, column_names)[source]

Get the transformed output space for a given input space.

Parameters
  • input_space (Space) – A gym.Space matching the shape of the pipeline’s input.

  • column_names (List[str]) – A list of all column names in the input data frame.

Return type

Space

Returns

A gym.Space matching the shape of the pipeline’s output.