def __create_variables()

in frauddetector/profiler.py [0:0]

• 16 lines of code
• 4 McCabe index (conditional complexity)

    def __create_variables(self, df_stats, event_column, timestamp_column):
        """Create variables for AFD
            
            Args:
                df_stats (pandas.core.frame.DataFrame): DataFrame of summary statistics, training data schema, event variables and event lables
                event_column (str): column that contains the target event
            Returns:
                variables (list): List of dicts with variable names
        """
        variables = []
        for i in range(df_stats.shape[0]):
            if df_stats.loc[i, "feature_name"] not in [event_column, timestamp_column]:
                data_type = "STRING"
                default_value = "unknown"
                if df_stats.loc[i, "feature_type"] == "NUMERIC":
                    data_type = "FLOAT"
                    default_value = 0.0
                variables.append({
                    "name": str(df_stats.loc[i, "feature_name"]),
                    "variableType": df_stats.loc[i, "feature_type"],
                    "dataType": data_type,
                    "defaultValue": "unknown"
                })
        return variables