Add uniform and Gaussian weight initializers

initwfn/Gaussian.go

@@ -0,0 +1,31 @@
+package initwfn
+
+import G "gorgonia.org/gorgonia"
+
+// Gaussian implements a configuration of a weight initializer that
+// draws weights from a gaussian distribution
+type GaussianConfig struct {
+	Mean, StdDev float64
+}
+
+// NewGaussian returns a new gaussian weight initializer
+func NewGaussian(mean, stddev float64) (*InitWFn, error) {
+	config := GaussianConfig{
+		Mean:   mean,
+		StdDev: stddev,
+	}
+
+	return newInitWFn(config)
+}
+
+// Type returns the type of initialization algorithm described by
+// the configuration.
+func (u GaussianConfig) Type() Type {
+	return Gaussian
+}
+
+// Create returns the weight initialization algorithm as a Gorgonia
+// InitWFn
+func (u GaussianConfig) Create() G.InitWFn {
+	return G.Gaussian(u.Mean, u.StdDev)
+}

initwfn/InitWFn.go

@@ -23,6 +23,8 @@ const (
 	Zeroes   Type = "Zeroes"
 	Ones     Type = "Ones"
 	Constant Type = "Constant"
+	Uniform  Type = "Uniform"
+	Gaussian Type = "Gaussian"
 )
 
 // InitWFn wraps Gorgonia InitWFn so that they can be JSON marshalled and
@@ -65,6 +67,8 @@ func (i *InitWFn) UnmarshalJSON(data []byte) error {
 			string(Zeroes):   reflect.TypeOf(ZeroesConfig{}),
 			string(Ones):     reflect.TypeOf(OnesConfig{}),
 			string(Constant): reflect.TypeOf(ConstantConfig{}),
+			string(Uniform):  reflect.TypeOf(UniformConfig{}),
+			string(Gaussian): reflect.TypeOf(GaussianConfig{}),
 		})
 	if err != nil {
 		return err

initwfn/Uniform.go

@@ -0,0 +1,31 @@
+package initwfn
+
+import G "gorgonia.org/gorgonia"
+
+// Uniform implements a configuration of a weight initializer that
+// draws weights from a uniform distribution
+type UniformConfig struct {
+	Low, High float64
+}
+
+// NewUniform returns a new uniform weight initializer
+func NewUniform(low, high float64) (*InitWFn, error) {
+	config := UniformConfig{
+		Low:  low,
+		High: high,
+	}
+
+	return newInitWFn(config)
+}
+
+// Type returns the type of initialization algorithm described by
+// the configuration.
+func (u UniformConfig) Type() Type {
+	return Uniform
+}
+
+// Create returns the weight initialization algorithm as a Gorgonia
+// InitWFn
+func (u UniformConfig) Create() G.InitWFn {
+	return G.Uniform(u.Low, u.High)
+}