def __init__()

in sagemaker/source/dl_utils/dataset.py [0:0]

• 15 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(self, file, sensor_headers, target_column, standardize=True):      
        super().__init__()
        self.target_column = target_column
        self.sensor_headers = sensor_headers
        
        print('Creating PMDataset:', file)
        print('  * Loading CSV data')
        df = pd.read_csv(file)
        df = df.dropna()
        
        self.vehicle_properties_headers = self._get_vehicle_properties_headers(df)
        
        self._check_sensor_headers(df)
                
        vehicle_properties_df, sensor_df = self._split_df_columns(df)
        
        mean_dict = self._get_means(sensor_df)
        
        self.data, self.labels = self._build_sensor_output_data(sensor_df, standardize, mean_dict)
        self.vehicle_properties = vehicle_properties_df

        print("Done")