tensortrade.pipeline.transformers.delete_columns module

class tensortrade.pipeline.transformers.delete_columns.DeleteColumnsTransformer(columns: List[str] = None)

Bases: AbstractTransformer

Initializes the delete columns transformer.

Parameters:

columns (List[str]) – List of columns to delete.

transform(df: DataFrame) → DataFrame

Deletes the defined columns of the DataFrames.

Parameters:

df (DataFrame) – The dataframe to delete the columns from.

Returns:

The dataframe with the columns deleted.

Return type:

DataFrame