Skip to content

Commit

Permalink
psobat numpy int serialization fix
Browse files Browse the repository at this point in the history
  • Loading branch information
gykovacs committed Aug 21, 2022
1 parent ec7e0ac commit ea79349
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion smote_variants/_version.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,4 +6,4 @@
@author: gykovacs
"""

__version__= '0.6.4'
__version__= '0.6.5'
2 changes: 1 addition & 1 deletion smote_variants/oversampling/_smote_psobat.py
Original file line number Diff line number Diff line change
Expand Up @@ -775,7 +775,7 @@ def sampling_algorithm(self, X, y):
elif self.method == 'bat':
best_combination = self.bat_vectorized(X, y, nn_params)

ss_params = self.ss_params
ss_params = self.ss_params.copy()
ss_params['n_dim'] = np.min([ss_params['n_dim'], int(best_combination[0])])

return SMOTE(proportion=best_combination[1],
Expand Down

0 comments on commit ea79349

Please sign in to comment.