|
|
@@ -11,7 +11,7 @@ const (
|
|
|
Input int = 1433
|
|
|
Hidden int = 64
|
|
|
Output int = 7
|
|
|
- Sample int = 10000
|
|
|
+ Sample int = 10000000
|
|
|
Batch int = 5
|
|
|
Dropout float64 = 0.5
|
|
|
Rate float64 = 0.1
|
|
|
@@ -122,7 +122,7 @@ func Train(G Graph) []Layer {
|
|
|
p2 := Parameter{W: MakeRandomMatrix(Hidden, Output), B: MakeRandomMatrix(Hidden, Output)}
|
|
|
l := []Layer{{d: Input, p: p0}, {d: Hidden, f: ReLU, p: p1}, {d: Output, f: Softmax, p: p2}}
|
|
|
for i := 0; i < Sample; i++ {
|
|
|
- if i%100 == 0 {
|
|
|
+ if i%1000 == 0 {
|
|
|
fmt.Println("sampling", i)
|
|
|
}
|
|
|
var wg sync.WaitGroup
|
