我将有关训练模型的信息存储在 DataFrame 中:
score time
LogisticRegression 1.0 0.257812
RidgeClassifier 1.0 0.283841
RandomForestClassifier 1.0 0.569159
GradientBoostingClassifier 1.0 9.408432
我需要找到模型的索引(名称),它具有最高的分数和最短的时间。我怎么做?
sort_values
一侧上升,另一侧下降:
>>> df.sort_values(["score", "time"], ascending=[False, True]).index[0]
"LogisticRegression"
并在结果中获取索引中的第 0 个值。
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句