断言两个 sklearn 对象相等

Question

My question is simple and divided in two sub-questions :

• I have two LabelEncoder s with the exact same parameter .classes_ . But when I compare then with == , they seems different. How can I check they are equal ?

• I also want to check two sklearn model ( RandomForest already fitted) are equals ? The == check does not seems to work.

Answer 1

If you compare two objects with ==, it will return False because they have different id.

You'd better override __eq__ function of class.

Answer 2

As mentioned in the comments, we need to create our own condition for __eq__ . Here is my version:

from sklearn.preprocessing import LabelEncoder
import numpy as np
target = pd.Series(np.random.choice(['yes', 'no'], (20,)))

class MyLabelEncoder(LabelEncoder):
    def __eq__(self, other):
        if np.array_equal(other.classes_,
                                   self.classes_):
            return True
        return False

le1 = MyLabelEncoder().fit(target)
le2 = MyLabelEncoder().fit(target)
le1 == le2
# True

For DecisionTree, there is already a solution provided here . You could extend it for RandomForestClassifier .