ucb

Documentation

Index

• type UCB
  • func New(arms int, attempts, rewards []int) (*UCB, error)
  • func (u *UCB) SelectArm() (int, error)

Types

type UCB

type UCB struct {
	sync.RWMutex
	Arms     int
	Attempts []int
	Rewards  []int
}

UCB represents UCB multi-arm bandit.

func New

func New(arms int, attempts, rewards []int) (*UCB, error)

New creates a new UCB multi-arm bandit.

func (*UCB) SelectArm

func (u *UCB) SelectArm() (int, error)

SelectArm selects arm to trigger multi-arm bandit.