Ejemplo n.º 1
0
 def mkNet(self, margin=1, dropout=0.5):
     f = NetFactory(self.config)
     netname = f.init("NetStockDaysWithProfit", {
         "margin": margin,
         "dropout": dropout
     }, "gpu%d_" % self.gpu_id + "alex.prototxt")
     return netname
Ejemplo n.º 2
0
 def mkNet(self, margin=1, dropout=0.5):
     f = NetFactory(self.config)
     netname = f.init("TestNet", {
         "margin": margin,
         "dropout": dropout
     }, "gpu%d_" % self.gpu_id + "test.prototxt")
     return netname
Ejemplo n.º 3
0
 def mkNet(self, margin=1, dropout=0.5):
     f = NetFactory(self.config)
     netname = f.init("StockInception", {
         "margin": margin,
         "dropout": dropout
     }, "gpu%d_" % self.gpu_id + "inception.prototxt")
     return netname
Ejemplo n.º 4
0
 def mkNet(self, margin=1):
     f = NetFactory(self.config)
     f.init("Net", {"margin": margin})
     return