开发者ID:apyeh,项目名称:UnbalancedDataset,代码行数:11,代码来源:test_smote_enn.py fitdeftest_smote_fit():"""Test the fitting method"""# Create the objectsmote = SMOTEENN(random_state=RND_SEED)# Fit the datasmote.fit(X, Y)# Check if the data information have been computedassert_equal(...
# 需要导入模块: from imblearn import combine [as 别名]# 或者: from imblearn.combine importSMOTEENN[as 别名]defcreate_sampler(sampler_name, random_state=None):ifsampler_nameisNoneorsampler_name =='None':returnNoneifsampler_name.lower() =='randomundersampler':returnRandomUnderSampler(random_state...
# 需要导入模块: from imblearn.combine import SMOTEENN [as 别名]# 或者: from imblearn.combine.SMOTEENN importfit_resample[as 别名]deftest_error_wrong_object():smote ='rnd'enn ='rnd'smt = SMOTEENN(smote=smote, random_state=RND_SEED)withraises(ValueError, match="smote needs to be a SMOTE"...
# 需要导入模块: from imblearn.combine import SMOTEENN [as 别名]# 或者: from imblearn.combine.SMOTEENN importfit_sample[as 别名]deftest_sample_regular():"""Test sample function with regular SMOTE."""# Create the objectsmote = SMOTEENN(random_state=RND_SEED)# Fit the datasmote.fit(X, Y)...