def test():
testing = Testing()
testModelBuilder(testing)
if testing.DidTestFail():
return 1
else:
return 0
def test():
testing = Testing()
hingeLossTest(testing)
logLossTest(testing)
squaredLossTest(testing)
if testing.DidTestFail():
return 1
else:
return 0
def test():
testing = Testing()
# -dd auto -sw 1 -sb 1 -sz 1 -pd 10 -l 2 -mp 5 -v --evaluationFrequency 1 -plf L2
args = ell.ProtoNNTrainerParameters()
args.projectedDimension = 10
args.numPrototypesPerLabel = 5
args.numLabels = 2
args.sparsityW = 1
args.sparsityB = 1
args.sparsityZ = 1
args.gamma = -1
args.lossFunction = ell.ProtoNNLossFunction.L2
args.numInnerIterations = 1
args.numFeatures = 0
args.verbose = True
trainer = ell.ProtoNNTrainer(args)
dataset = ell.AutoSupervisedDataset()
testFile = os.path.join(find_ell.get_ell_root(),
"examples/data/protonnTestData.txt")
print("Loading: " + testFile)
dataset.Load(testFile)
total = dataset.NumExamples()
features = dataset.NumFeatures()
testing.ProcessTest("Proton dataset loaded",
testing.IsEqual(int(total), 200))
trainer.SetDataset(dataset)
numIterations = 20
print("Training...")
for i in range(numIterations):
trainer.Update()
predictor = trainer.GetPredictor()
accuracy = get_accuracy(predictor, dataset, features)
print("Accuracy %f" % (accuracy))
testing.ProcessTest("Proton accuracy test",
testing.IsEqual(int(accuracy), 1))
map = predictor.GetMap()
map.Save("protonnTestData.ell")
testing.ProcessTest(
"Saving protonnTestData.ell",
testing.IsEqual(os.path.isfile("protonnTestData.ell"), True))
if testing.DidTestFail():
raise Exception("protonn_trainer_test failed")
return 0
def test():
# this test only tests the model interface. If you are testing node types then use nodes_test.py
testing = Testing()
test_callbacks(testing)
test_multiple(testing)
test_bitcode(testing)
if testing.DidTestFail():
return 1
else:
return 0
def test():
testing = Testing()
test_callbacks(testing)
test_typecast(testing)
test_unary(testing)
test_multiple(testing)
#test_multiply(testing) # bugbug: crashing on Linux...
test_bitcode(testing)
if testing.DidTestFail():
return 1
else:
return 0
def test():
testing = Testing()
# -dd auto -sw 1 -sb 1 -sz 1 -pd 10 -l 2 -mp 5 -v --evaluationFrequency 1 -plf L2
args = ell.trainers.ProtoNNTrainerParameters()
args.projectedDimension = 10
args.numPrototypesPerLabel = 5
args.numLabels = 2
args.sparsityW = 1
args.sparsityB = 1
args.sparsityZ = 1
args.gamma = -1
args.lossFunction = ell.trainers.ProtoNNLossFunction.L2
args.numInnerIterations = 1
args.numFeatures = 0
args.verbose = True
trainer = ell.trainers.ProtoNNTrainer(args)
dataset = ell.data.AutoSupervisedDataset()
testFile = os.path.join(find_ell.get_ell_root(),
"examples/data/protonnTestData.txt")
print("Loading: " + testFile)
dataset.Load(testFile)
total = dataset.NumExamples()
features = dataset.NumFeatures()
testing.ProcessTest("ProtoNN dataset loaded",
testing.IsEqual(int(total), 200))
trainer.SetDataset(dataset)
numIterations = 20
print("Training...")
for i in range(numIterations):
trainer.Update()
predictor = trainer.GetPredictor()
accuracy = get_accuracy(predictor, dataset, features, predictor.Predict)
print("Accuracy %f" % (accuracy))
testing.ProcessTest("ProtoNN accuracy test",
testing.IsEqual(int(accuracy), 1))
map = predictor.GetMap()
map.Save("protonnTestData.ell")
testing.ProcessTest(
"Saving protonnTestData.ell",
testing.IsEqual(os.path.isfile("protonnTestData.ell"), True))
# make sure we can compile this map.
try:
compilerSettings = ell.model.MapCompilerOptions()
compilerSettings.useBlas = False
optimizerSettings = ell.model.ModelOptimizerOptions()
compiledMap = map.CompileDouble("host", "protonn", "predict",
compilerSettings, optimizerSettings)
compiled = True
except Exception as e:
print("Compile ProtoNN model failed: {}", e)
compiled = False
testing.ProcessTest("Compiling protonnTestData.ell",
testing.IsEqual(compiled, True))
if compiled:
accuracy = get_accuracy(predictor, dataset, features,
compiledMap.ComputeDouble)
print("Compiled Accuracy %f" % (accuracy))
testing.ProcessTest("ProtoNN compiled accuracy test",
testing.IsEqual(int(accuracy), 1))
if testing.DidTestFail():
raise Exception("protonn_trainer_test failed")
return 0