(self, random_state: int = 42)
| 80 | |
| 81 | |
| 82 | def __init__(self, random_state: int = 42): |
| 83 | self.random_state = random_state |
| 84 | self.models = { |
| 85 | 'random_forest': RandomForestClassifier(random_state=random_state), |
| 86 | 'gradient_boosting': GradientBoostingClassifier(random_state=random_state) |
| 87 | } |
| 88 | self.best_model = None |
| 89 | self.best_score = float('-inf') |
| 90 | |
| 91 | def select_best_model(self, X_train: np.ndarray, y_train: np.ndarray, |
| 92 | X_val: np.ndarray, y_val: np.ndarray) -> BaseEstimator: |
nothing calls this directly
no outgoing calls
no test coverage detected