ml

Documentation

Index

• type BaseEstimator
  • func NewSimpleKNN(params map[string]interface{}) BaseEstimator
• type Item
• type MaxHeapK
  • func NewMaxHeapK(k int) *MaxHeapK
  • func (pq *MaxHeapK) Add(x interface{})
  • func (pq MaxHeapK) Len() int
  • func (pq MaxHeapK) Less(i, j int) bool
  • func (pq *MaxHeapK) Pop() interface{}
  • func (pq *MaxHeapK) Push(x interface{})
  • func (pq MaxHeapK) Swap(i, j int)
• type SimpleKNN
  • func (knn *SimpleKNN) Fit(activities map[string]map[string]int64)
  • func (knn *SimpleKNN) Predict(userID, channelID string) (float64, error)
  • func (knn *SimpleKNN) SetParams(params map[string]interface{})

Types

type BaseEstimator

type BaseEstimator interface {
	SetParams(params map[string]interface{})
	Predict(userID, channelID string) (float64, error)
	Fit(activities map[string]map[string]int64)
}

BaseEstimator determines interface for all estimators for user-channel suggestions

func NewSimpleKNN

func NewSimpleKNN(params map[string]interface{}) BaseEstimator

NewSimpleKNN returns Simple KNN Estimator

type Item

type Item struct {
	// contains filtered or unexported fields
}

An Item is something we manage in a max heap.

type MaxHeapK

type MaxHeapK struct {
	// contains filtered or unexported fields
}

A MaxHeapK implements heap.Interface.

func NewMaxHeapK

func NewMaxHeapK(k int) *MaxHeapK

NewMaxHeapK returns a brand new max heap with capacity k.

func (*MaxHeapK) Add

func (pq *MaxHeapK) Add(x interface{})

Add method adds element into the PQ.

func (MaxHeapK) Len

func (pq MaxHeapK) Len() int

Len returns number of elements in a heap

func (MaxHeapK) Less

func (pq MaxHeapK) Less(i, j int) bool

Less implements comparison function for a heap

func (*MaxHeapK) Pop

func (pq *MaxHeapK) Pop() interface{}

Pop method pops item of hightest priority from the PQ.

func (*MaxHeapK) Push

func (pq *MaxHeapK) Push(x interface{})

Push method pushes an item into the PQ.

func (MaxHeapK) Swap

func (pq MaxHeapK) Swap(i, j int)

Swap swaps two elements of a heap

type SimpleKNN

type SimpleKNN struct {
	// contains filtered or unexported fields
}

SimpleKNN struct

func (*SimpleKNN) Fit

func (knn *SimpleKNN) Fit(activities map[string]map[string]int64)

Fit the KNN estimator

func (*SimpleKNN) Predict

func (knn *SimpleKNN) Predict(userID, channelID string) (float64, error)

Predict the activity of channel channelID for userID

func (*SimpleKNN) SetParams

func (knn *SimpleKNN) SetParams(params map[string]interface{})

SetParams sets parameters for KNN estimator