Skip to content
Snippets Groups Projects
Commit beb78210 authored by Baptiste Bauvin's avatar Baptiste Bauvin
Browse files

Time accuracy

parent 4012c1b9
No related branches found
No related tags found
No related merge requests found
...@@ -75,13 +75,14 @@ class ColumnGenerationClassifierQar(BaseEstimator, ClassifierMixin, BaseBoost): ...@@ -75,13 +75,14 @@ class ColumnGenerationClassifierQar(BaseEstimator, ClassifierMixin, BaseBoost):
def fit(self, X, y): def fit(self, X, y):
start = time.time()
formatted_X, formatted_y = self.format_X_y(X, y) formatted_X, formatted_y = self.format_X_y(X, y)
self.init_info_containers() self.init_info_containers()
m, n, y_kernel_matrix = self.init_hypotheses(formatted_X, formatted_y) m, n, y_kernel_matrix = self.init_hypotheses(formatted_X, formatted_y)
start = time.time()
self.n_total_hypotheses_ = n self.n_total_hypotheses_ = n
self.n_total_examples = m self.n_total_examples = m
......
...@@ -39,8 +39,8 @@ class AdaboostPregenTree(AdaBoostClassifier, BaseMonoviewClassifier, PregenClass ...@@ -39,8 +39,8 @@ class AdaboostPregenTree(AdaBoostClassifier, BaseMonoviewClassifier, PregenClass
self.random_state = random_state self.random_state = random_state
def fit(self, X, y, sample_weight=None): def fit(self, X, y, sample_weight=None):
begin = time.time()
pregen_X, pregen_y = self.pregen_voters(X, y, generator=self.estimators_generator) pregen_X, pregen_y = self.pregen_voters(X, y, generator=self.estimators_generator)
begin = time.time()
super(AdaboostPregenTree, self).fit(pregen_X, pregen_y, sample_weight=sample_weight) super(AdaboostPregenTree, self).fit(pregen_X, pregen_y, sample_weight=sample_weight)
end = time.time() end = time.time()
self.train_time = end-begin self.train_time = end-begin
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment