Class Pegasos

constructor

• new Pegasos(rate: number, k?: number): Pegasos

• Parameters

  • rate: number

    Learning rate
  • Optionalk: number

    Batch size

  Returns Pegasos

_b

_do_projection

_itr

_k

_r

_t

_w

_x

_y

fit

• fit(): void

• Fit model parameters.

  Returns void

init

• init(train_x: number[][], train_y: (-1 | 1)[]): void

• Initialize this model.

  Parameters

  • train_x: number[][]

    Training data
  • train_y: (-1 | 1)[]

    Target values

  Returns void

predict

• predict(data: number[][]): (-1 | 1)[]

• Returns predicted values.

  Parameters

  • data: number[][]

    Sample data

  Returns (-1 | 1)[]

  Predicted values

update

• update(x: number[][], y: (-1 | 1)[]): void

• Update model parameters with some data.

  Parameters

  • x: number[][]

    Training data
  • y: (-1 | 1)[]

    Target value

  Returns void