logaritmisk/trueskill-tt
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

More rustifying

Browse Source
This commit is contained in:
Anders Olsson
2022-12-27 22:30:20 +01:00
parent 9b6cb9e7eb
commit 8e25826f91
1 changed files with 27 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
src/batch.rs
View File

@@ -110,7 +110,7 @@ pub struct Batch {
impl Batch {
pub fn new(
compositions: Vec<Vec<Vec<Index>>>,
composition: Vec<Vec<Vec<Index>>>,
results: Vec<Vec<f64>>,
weights: Vec<Vec<Vec<f64>>>,
time: i64,
@@ -118,15 +118,15 @@ impl Batch {
agents: &HashMap<Index, Agent>,
) -> Self {
assert!(
results.is_empty() || results.len() == compositions.len(),
results.is_empty() || results.len() == composition.len(),
"TODO: Add a comment here"
);
assert!(
weights.is_empty() || weights.len() == compositions.len(),
weights.is_empty() || weights.len() == composition.len(),
"TODO: Add a comment here"
);
let this_agent = compositions
let this_agent = composition
.iter()
.flatten()
.flatten()
@@ -148,13 +148,13 @@ impl Batch {
})
.collect::<HashMap<_, _>>();
let events = compositions
let events = composition
.iter()
.enumerate() // (0..composition.len())
.map(|(e, composition)| {
let teams = composition
.enumerate()
.map(|(e, event)| {
let teams = event
.iter()
.enumerate() // (0..composition.len())
.enumerate()
.map(|(t, team)| {
let items = team
.iter()
@@ -167,7 +167,7 @@ impl Batch {
Team {
items,
output: if results.is_empty() {
(composition.len() - (t + 1)) as f64
(event.len() - (t + 1)) as f64
} else {
results[e][t]
},
@@ -231,14 +231,15 @@ impl Batch {
}
}
let from = self.events.len();
for e in 0..composition.len() {
let teams = (0..composition[e].len())
.map(|t| {
let items = (0..composition[e][t].len())
.map(|a| Item {
agent: composition[e][t][a],
let events = composition.iter().enumerate().map(|(e, event)| {
let teams = event
.iter()
.enumerate()
.map(|(t, team)| {
let items = team
.iter()
.map(|&agent| Item {
agent,
likelihood: N_INF,
})
.collect::<Vec<_>>();
@@ -246,7 +247,7 @@ impl Batch {
Team {
items,
output: if results.is_empty() {
(composition[e].len() - (t + 1)) as f64
(event.len() - (t + 1)) as f64
} else {
results[e][t]
},
@@ -254,7 +255,7 @@ impl Batch {
})
.collect::<Vec<_>>();
let event = Event {
Event {
teams,
evidence: 0.0,
weights: if weights.is_empty() {
@@ -262,10 +263,12 @@ impl Batch {
} else {
weights[e].clone()
},
};
}
});
self.events.push(event);
}
let from = self.events.len();
self.events.extend(events);
self.iteration(from, agents);
}
@@ -615,7 +618,7 @@ mod tests {
],
vec![vec![1.0, 0.0], vec![0.0, 1.0], vec![1.0, 0.0]],
vec![],
&mut agents,
&agents,
);
assert_eq!(batch.events.len(), 6);