Skip to content

Commit

Permalink
fix experiments script
Browse files Browse the repository at this point in the history
  • Loading branch information
amatissart committed Jun 1, 2024
1 parent 5e6d598 commit 051f088
Showing 1 changed file with 9 additions and 3 deletions.
12 changes: 9 additions & 3 deletions solidago/experiments/tournesol.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,8 +38,6 @@
for entity_id, video_id in enumerate(inputs.entity_id_to_video_id)
}

logger.info("Preprocessing data for the pipeline")
users, vouches, all_entities, privacy = inputs.get_pipeline_objects()

# criteria = set(inputs.comparisons["criteria"])
criteria = { "largely_recommended" }
Expand Down Expand Up @@ -89,10 +87,18 @@

user_outputs, entities, voting_rights, scaled_user_models = dict(), dict(), dict(), dict()

users = pipeline.trust_propagation(users, vouches)

for c in criteria:
logger.info(f"Running the pipeline for criterion `{c}`")

pipeline_objects = inputs.get_pipeline_kwargs(criterion=c)
users = pipeline_objects["users"]
vouches = pipeline_objects["vouches"]
all_entities = pipeline_objects["entities"]
privacy = pipeline_objects["privacy"]
judgments = pipeline_objects["judgements"]

users = pipeline.trust_propagation(users, vouches)

judgments = inputs.get_judgments(c)

Expand Down

0 comments on commit 051f088

Please sign in to comment.