efficientcube

package

v0.5.0 Latest Latest Go to latest Published: Aug 7, 2026 License: MIT Imports: 9 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

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

func NewAgent(p *Policy, cfg BeamConfig) *Agent

func (a *Agent) Act(s core.State) core.Action

func (a *Agent) EndEpisode(total core.Reward)

func (a *Agent) Observe(s core.State, act core.Action, r core.Reward, next core.State, done bool)

type BeamConfig struct {
	Width    int
	MaxDepth int
}

func DefaultBeamConfig() BeamConfig

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

func LoadPolicy(path string) (*Policy, error)

func NewPolicy(hidden []int, seed int64) *Policy

func Train(cfg TrainConfig, logFn func(iter int, loss, acc float64)) *Policy

func (p *Policy) Best(c rubix.Cube) rubix.Move

func (p *Policy) Probs(c rubix.Cube) []float64

func (p *Policy) Save(path string) error

func (p *Policy) Solve(start rubix.Cube, cfg BeamConfig) Result

func (p *Policy) SolveBatch(cubes []rubix.Cube, cfg BeamConfig) []Result

type Result struct {
	Moves  []rubix.Move
	Solved bool
	Nodes  int
}

type TrainConfig struct {
	Hidden []int
	K      int
	Batch  int
	Iters  int
	LR     float64
	Seed   int64
}

func DefaultTrainConfig() TrainConfig