Struct NegTrain

pub struct NegTrain { /* private fields */ }

Builder for the NegTrain operation.

Implementations

impl NegTrain

pub fn new() -> Self

Creates a new NegTrain.

pub fn vocab_count<ArgType: Into<Vec<i64>>>(self, value: ArgType) -> Self

Sets the vocab_count attribute.

pub fn num_negative_samples<ArgType: Into<i64>>(self, value: ArgType) -> Self

Sets the num_negative_samples attribute.

pub fn add_control_input(self, op: Operation) -> Self

Adds a control input.

pub fn build<O0: Into<Output>, O1: Into<Output>, O2: Into<Output>, O3: Into<Output>, O4: Into<Output>>( &self, w_in: O0, w_out: O1, examples: O2, labels: O3, lr: O4, scope: &mut Scope, ) -> Result<Operation>

Builds the NegTrain operation.

Trait Implementations

impl Debug for NegTrain

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for NegTrain

fn default() -> NegTrain

Returns the “default value” for a type.