onehot 0.1.0

One-hot encoding trait and proc macro for auto deriving
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59

use bitmatrix::BitMatrix;

use crate::OneHot;


/// Encode a dataset vertically into a [bitmatrix](https://docs.rs/bitmatrix/) using OneHot.
///
/// The resulting matrix is of the form:
/// ```notrust
///          record 1  . . .  record N
///         +--------------------------
/// value 1 | true     . . .   false
///    .    |   .      . . .     .
///    .    |   .      . . .     .
///    .    |   .      . . .     .
/// value M | false    . . .   false
/// ```
pub fn vertical<T: OneHot>(data: &[T]) -> BitMatrix {
	let mut matrix = BitMatrix::new(
		T::ONEHOT_LEN,
		data.len(),
	);

	for (transaction, record) in data.iter().enumerate() {
		for (item, value) in record.onehot().enumerate() {
			matrix[item].set(transaction, value);
		}
	}

	matrix
}


/// Encode a dataset horizontally into a [bitmatrix](https://docs.rs/bitmatrix/) using OneHot.
///
/// The resulting matrix is of the form:
/// ```notrust
///            value 1    . . .  value N
///          +---------------------------
/// record 1 |  true      . . .   false
///    .     |    .       . . .     .
///    .     |    .       . . .     .
///    .     |    .       . . .     .
/// record M |  false     . . .   false
/// ```
pub fn horizontal<T: OneHot>(data: &[T]) -> BitMatrix {
	let mut matrix = BitMatrix::new(
		data.len(),
		T::ONEHOT_LEN,
	);

	for (transaction, record) in data.iter().enumerate() {
		for (item, value) in record.onehot().enumerate() {
			matrix[transaction].set(item, value);
		}
	}

	matrix
}