Skip to content

Commit 642a5d4

Browse files
author
Guillaume Lemaitre
committed
add logger in base class
1 parent 1a9a1b8 commit 642a5d4

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

imblearn/base.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
from __future__ import print_function
55

66
import warnings
7+
import logging
78

89
import numpy as np
910

@@ -54,8 +55,6 @@ def __init__(self, ratio='auto', verbose=True):
5455
self.ratio = ratio
5556
self.verbose = verbose
5657

57-
# Create the member variables regarding the classes statistics
58-
5958
def fit(self, X, y):
6059
"""Find the classes statistics before to perform sampling.
6160
@@ -74,6 +73,9 @@ def fit(self, X, y):
7473
7574
"""
7675

76+
# Create logger
77+
self.logger = logging.getLogger(__name__)
78+
7779
# Check the consistency of X and y
7880
X, y = check_X_y(X, y)
7981

0 commit comments

Comments
 (0)