fix: 移除旧版 sklearn 不兼容的 multi_class 参数
This commit is contained in:
parent
562fc4142a
commit
307b66e9db
1 changed files with 1 additions and 3 deletions
|
|
@ -64,9 +64,7 @@ class HOGLRBaseline:
|
||||||
self.data_root = data_root
|
self.data_root = data_root
|
||||||
self.image_size = image_size
|
self.image_size = image_size
|
||||||
self.clf = LogisticRegression(
|
self.clf = LogisticRegression(
|
||||||
C=1.0, max_iter=1000,
|
C=1.0, max_iter=1000, solver='lbfgs', n_jobs=-1,
|
||||||
multi_class='multinomial', solver='lbfgs',
|
|
||||||
n_jobs=-1,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
def _load_data(self, split):
|
def _load_data(self, split):
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue