naprawianie all

app.py

@@ -4,7 +4,7 @@ import pandas as pd
 import os
 # import matplotlib.pyplot as plt
 
-from utils import fetch_runs_to_df, fetch_run
+from utils import fetch_runs_to_df, fetch_run, fetch_models_to_df
 
 # Access the API key from the environment variable
 wandb_api_key = os.getenv('WANDB_API_KEY')
@@ -84,27 +84,8 @@ if option == "Models":
 
  # Ensure the DataFrame is not empty
  if not df.empty:
- # Fetch metrics for ranking (e.g., accuracy or loss)
- ranking_data = []
- for index, row in df.iterrows():
- try:
- # Fetch the run details
- run = api.run(f"{projects[selected_project]['entity']}/{projects[selected_project]['project']}/{row['ID']}")
- metrics = run.summary
- model_name = run.config.get("model_name", "Unknown") # Fetch model name from the config, defaulting to "Unknown"
- 
- ranking_data.append({
- "Model Name": model_name, # Add model name to the table
- "Run Name": row["Run Name"],
- "ID": row["ID"],
- "Accuracy": metrics.get("accuracy"), # Example metric
- "Loss": metrics.get("loss") # Example metric
- })
- except wandb.errors.CommError:
- continue
-
  # Convert to DataFrame
- ranking_df = pd.DataFrame(ranking_data)
+ ranking_df = fetch_models_to_df(api, projects, selected_project, df)
 
  # Rank by Accuracy (or another metric)
  ranking_df = ranking_df.sort_values(by="Accuracy", ascending=False).reset_index(drop=True)

utils.py

@@ -57,4 +57,43 @@ def fetch_run(api, projects, selected_project, selected_run_id):
  project = projects[selected_project]["project"]
  run = api.run(f"{entity}/{project}/{selected_run_id}")
 
- return run
+ return run
+
+
+def fetch_models_to_df(api, projects, selected_project, df):
+ data = []
+ for index, row in df.iterrows():
+ try:
+ if selected_project == "All":
+ # Determine the project for the current run
+ for project_name, details in projects.items():
+ entity = details["entity"]
+ project = details["project"]
+ try:
+ run = api.run(f"{entity}/{project}/{row['ID']}")
+ break
+ except wandb.errors.CommError:
+ continue
+ else:
+ st.error(f"Run ID {row['ID']} not found in any project.")
+ continue
+ else:
+ entity = projects[selected_project]["entity"]
+ project = projects[selected_project]["project"]
+ run = api.run(f"{entity}/{project}/{row['ID']}")
+
+ metrics = run.summary
+ model_name = run.config.get("model_name", "Unknown") # Fetch model name from the config, defaulting to "Unknown"
+ 
+ data.append({
+ "Model Name": model_name, # Add model name to the table
+ "Run Name": row["Run Name"],
+ "ID": row["ID"],
+ "Accuracy": metrics.get("accuracy"), # Example metric
+ "Loss": metrics.get("loss") # Example metric
+ })
+ except wandb.errors.CommError:
+ continue
+
+ data_df = pd.DataFrame(data)
+ return data_df