(self)
| 633 | |
| 634 | class TestPredictor(unittest.TestCase): |
| 635 | def _create_model(self): |
| 636 | m = model_helper.ModelHelper() |
| 637 | y = brew.fc( |
| 638 | m, |
| 639 | "data", |
| 640 | "y", |
| 641 | dim_in=4, |
| 642 | dim_out=2, |
| 643 | weight_init=("ConstantFill", dict(value=1.0)), |
| 644 | bias_init=("ConstantFill", dict(value=0.0)), |
| 645 | axis=0, |
| 646 | ) |
| 647 | m.net.AddExternalOutput(y) |
| 648 | return m |
| 649 | |
| 650 | # Use this test with a bigger model to see how using Predictor allows to |
| 651 | # avoid issues with low protobuf size limit in Python |
no test coverage detected