uyzhang / JCLIP

12 stars 4 forks source link

baseline_ft版本分类结果全部为相同的值 #2

Closed cherrysherryplus closed 6 months ago

cherrysherryplus commented 6 months ago

运行baseline之后,结果正常,且分类结果符合视觉感受;但是运行baseline_ft之后,所有测试图片的分类结果的前5个类别ID,都是相同的值,如下图所示。 image

uyzhang commented 6 months ago

我们这边测试没有问题,请检查是不是scikit-learn这个框架是不是安装的有问题。

cherrysherryplus commented 6 months ago

好的,请问你使用的scikit-learn版本是多少呢?另外,我这边换了一下LogisticRegression的solver参数变成solver="sag"可以收敛了,但是使用默认的solver="lbfgs"还是会有问题。