FinBen / app.py
黄济民
feature: add auto evaluation tab
a79e05b
raw
history blame
2.88 kB
from apscheduler.schedulers.background import BackgroundScheduler
import gradio as gr
import pandas as pd
# Load leaderboard data
leaderboard_df = pd.read_csv('leaderboard.csv')
leaderboard_auto_df = pd.read_csv('leaderboard_auto.csv')
# Constants
TITLE = "Financial Natural Language Understanding and Prediction Evaluation Benchmark (FLARE) Leaderboard"
INTRODUCTION_TEXT = "The leaderboard shows the performance of various models in financial natural language understanding and prediction tasks."
COLS = [
("Model", "str"),
("FPB-acc", "number"),
("FPB-F1", "number"),
("FiQA-SA-F1", "number"),
("Headline-AvgF1", "number"),
("NER-EntityF1", "number"),
("FinQA-EmAcc", "number"),
("ConvFinQA-EmAcc", "number"),
("BigData22-Acc", "number"),
("BigData22-MCC", "number"),
("ACL18-Acc", "number"),
("ACL18-MCC", "number"),
("CIKM18-Acc", "number"),
("CIKM18-MCC", "number")
]
COLS_AUTO = [
("Model", "str"),
("FPB-acc", "number"),
("FPB-F1", "number"),
("FPB-missing", "number"),
("FiQA-SA-F1", "number"),
("FiQA-SA-missing", "number"),
("Headline-AvgF1", "number"),
("NER-EntityF1", "number"),
("FinQA-EmAcc", "number"),
("BigData22-Acc", "number"),
("BigData22-MCC", "number"),
("BigData22-missing", "number"),
("ACL18-Acc", "number"),
("ACL18-MCC", "number"),
("ACL18-missing", "number"),
("CIKM18-Acc", "number"),
("CIKM18-MCC", "number"),
("CIKM18-missing", "number"),
("FOMC-acc", "number"),
("FOMC-F1", "number"),
("FOMC-missing", "number"),
("FinerOrd-EntityF1", "number"),
("FinerOrd-F1", "number"),
("German-Acc", "number"),
("German-MCC", "number"),
("German-missing", "number"),
("Australian-Acc", "number"),
("Australian-MCC", "number"),
("Australian-missing", "number")
]
TYPES = [col_type for _, col_type in COLS]
TYPES_AUTO = [col_type for _, col_type in COLS_AUTO]
def create_leaderboard_table(df, headers, types):
return gr.components.Dataframe(
value=df.values.tolist(),
headers=[col_name for col_name, _ in headers],
datatype=types,
max_rows=10,
)
def dummy_fn():
pass
def launch_gradio():
iface = gr.Interface(
fn=dummy_fn,
layout="tab",
inputs=None,
outputs=None,
title=TITLE,
description=INTRODUCTION_TEXT,
tabs=["Self-hosted evaluation", "Automatic evaluation"],
)
with iface.tab("Self-hosted evaluation"):
create_leaderboard_table(leaderboard_df, COLS, TYPES)
with iface.tab("Automatic evaluation"):
create_leaderboard_table(leaderboard_auto_df, COLS_AUTO, TYPES_AUTO)
iface.launch()
scheduler = BackgroundScheduler()
scheduler.add_job(launch_gradio, "interval", seconds=3600)
scheduler.start()
# Launch immediately
launch_gradio()