Module mls_lib.feature_engineering.column_select

ColumnSelect: Component that selects columns from the input table.

Classes

class ColumnSelect (input_table, columns)

ColumnSelect: Component that selects columns from the input table.

Expand source code

class ColumnSelect(FeatureEngineeringStep):
    """ ColumnSelect: Component that selects columns from the input table. """
    def __init__(self, input_table, columns):
        super().__init__(
            input_table = input_table
        )
        self.columns = columns

    def execute(self):
        dataframe = self._get_input("input_table")

        data = dataframe.get_data()
        data = data[self.columns]

        new_df = DataFrame()
        new_df.set_data(data)

        self._set_output("resulting_table", new_df)

        self._finish_execution()

Ancestors

• FeatureEngineeringStep
• Step

Inherited members

• FeatureEngineeringStep:
  • execute
  • get_output
  • is_finished
  • is_ready