cave.utils.convert_for_epm module

cave.utils.convert_for_epm.convert_data_for_epm(scenario: smac.scenario.scenario.Scenario, runhistory: smac.runhistory.runhistory.RunHistory, logger=None)[source]

converts data from runhistory into EPM format

Parameters
  • scenario (Scenario) – smac.scenario.scenario.Scenario Object

  • runhistory (RunHistory) – smac.runhistory.runhistory.RunHistory Object with all necessary data

Returns

  • X (np.array) – X matrix with configuartion x features for all observed samples

  • y (np.array) – y matrix with all observations

  • types (np.array) – types of X cols – necessary to train our RF implementation