github.com/gorgonia/agogo@v0.1.1/dummy.go (about)

     1  package agogo
     2  
     3  import "github.com/gorgonia/agogo/game"
     4  
     5  type dummyInferer struct {
     6  	outputSize    int
     7  	currentPlayer game.Player
     8  }
     9  
    10  func (d dummyInferer) Infer(a []float32) (policy []float32, value float32, err error) {
    11  	switch d.currentPlayer {
    12  	case 1:
    13  		value = 1
    14  	case 2:
    15  		value = -1
    16  	}
    17  	policy = make([]float32, d.outputSize)
    18  	for i := range policy {
    19  		policy[i] = 1 / float32(d.outputSize)
    20  	}
    21  
    22  	return policy, value, nil
    23  }
    24  
    25  func (d dummyInferer) Close() error { return nil }