fix: 移除旧版 sklearn 不兼容的 multi_class 参数
This commit is contained in:
@@ -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):
|
||||||
|
|||||||
Reference in New Issue
Block a user