ANL-CEEESA/MIPLearn
1
0
Fork 0
mirror of https://github.com/ANL-CEEESA/MIPLearn.git synced 2025-12-07 18:08:51 -06:00
Code Issues Packages Projects Releases Wiki Activity

Move feature classes to features.py

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-04-05 20:38:31 -05:00
parent b11779817a
commit 6e326d5d6e
13 changed files with 71 additions and 70 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
miplearn/components/component.py
View File

@@ -7,7 +7,8 @@ from typing import Any, List, TYPE_CHECKING, Tuple, Dict, Hashable
import numpy as np
from miplearn.instance import Instance
from miplearn.types import LearningSolveStats, TrainingSample, Features
from miplearn.types import LearningSolveStats
from miplearn.features import TrainingSample, Features
if TYPE_CHECKING:
from miplearn.solvers.learning import LearningSolver
@@ -49,7 +50,7 @@ class Component:
their own statistics here. For example, PrimalSolutionComponent adds
statistics regarding the number of predicted variables. All statistics in
this dictionary are exported to the benchmark CSV file.
features: Features
features: miplearn.features.Features
Features describing the model.
training_data: TrainingSample
A dictionary containing data that may be useful for training machine