Struct sbr::data::Interactions

pub struct Interactions { /* fields omitted */ }

A collection of individual interactions.

Methods

impl Interactions

pub fn new(num_users: usize, num_items: usize) -> Self

Crate a new interactions object.

pub fn push(&mut self, interaction: Interaction)

Add a new interaction.

pub fn data(&self) -> &[Interaction]

Return the underlying data.

pub fn len(&self) -> usize

Give the number of contained interactions.

pub fn shuffle<R: Rng>(&mut self, rng: &mut R)

Shuffle the interactions in-place.

pub fn split_at(&self, idx: usize) -> (Self, Self)

Split interactions at idx.

pub fn split_by<F: Fn(&Interaction) -> bool>(&self, func: F) -> (Self, Self)

Split interactions by predicate.

pub fn to_triplet(&self) -> TripletInteractions

Covert to triplet representation.

pub fn to_compressed(&self) -> CompressedInteractions

Convert to compressed representation.

pub fn num_users(&self) -> usize

Return number of users.

pub fn num_items(&self) -> usize

Return number of items.

pub fn shape(&self) -> (usize, usize)

Return (num_users, num_items).

Trait Implementations

impl From<Vec<Interaction>> for Interactions

fn from(data: Vec<Interaction>) -> Interactions

Performs the conversion.

impl<'a> From<&'a Interactions> for CompressedInteractions

fn from(interactions: &Interactions) -> CompressedInteractions

Performs the conversion.

impl<'a> From<&'a Interactions> for TripletInteractions

fn from(interactions: &'a Interactions) -> Self

Performs the conversion.