python - 我需要一个函数来选择 Python 数据框中包含的最佳模型

Question

我有一个包含 3 列的数据框：模型、因变量 (y) 和 f1_score。

我需要一个函数来为每个因变量（y）选择最佳模型。最好的模型是 f1_score 值最大的模型。

你能帮我吗？

谢谢

标签： pythonfunctiondataframemodel

Answer 1

df = pd.DataFrame({"model": ["a", "b", "a", "c"], "feature": [1, 1, 2,2], "f1":[0.95, 0.98, 0.7, 0.75]})
pd.pivot_table(df, index="feature", values="model", aggfunc=max)
        model
feature
1           b
2           c