bulletformat 1.8.0

Binary Data Formats, Data Loader and Utilities for bullet.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

mod ataxx;
pub mod chess;
mod convert;
mod loader;
mod util;

use std::{
    fs::File,
    io::{self, BufWriter, Write},
    marker::Sized,
};

pub use ataxx::AtaxxBoard;
pub use chess::ChessBoard;
pub use convert::{convert_from_bin, convert_from_text};
pub use loader::DataLoader;

pub trait BulletFormat: IntoIterator + Sized + Copy + Send + Sync {
    type FeatureType;

    const HEADER_SIZE: usize;

    fn set_result(&mut self, result: f32);

    fn score(&self) -> i16;

    fn result(&self) -> f32;

    fn result_idx(&self) -> usize;

    fn blended_result(&self, blend: f32, scale: f32) -> f32 {
        blend * self.result() + (1. - blend) * util::sigmoid(f32::from(self.score()), scale)
    }

    fn as_bytes_slice(data: &[Self]) -> &[u8] {
        util::to_slice_with_lifetime(data)
    }

    fn write_to_bin(output: &mut BufWriter<File>, data: &[Self]) -> io::Result<()> {
        let data_slice = util::to_slice_with_lifetime(data);
        output.write_all(data_slice)?;
        Ok(())
    }
}