rskit_dataset/label.rs
1use serde::{Deserialize, Serialize};
2
3/// Binary classification label.
4#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, Serialize, Deserialize)]
5#[non_exhaustive]
6#[repr(u8)]
7pub enum Label {
8 /// Human-authored / real sample.
9 Real = 0,
10 /// AI-generated sample.
11 AiGenerated = 1,
12}
13
14impl std::fmt::Display for Label {
15 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
16 match self {
17 Label::Real => write!(f, "real"),
18 Label::AiGenerated => write!(f, "ai"),
19 }
20 }
21}