engine_tests.py 文件源码

def test_simple_engine():
    engine = Engine({
        'datasource': {
            'class': TestDataSource,
            'params': {
                'csv': test_data_file,
            },
        },
        'algorithm': {
            'class': TestSimpleAlgorithm,
            'params': {
                'model.pickle': '~/.tidml/tests/model.pkl',  # default built-in
            },
        },
    })
    engine.train()
    models = engine.load_models()
    prediction = engine.predict(models, 3)
    nt.assert_equals(prediction, 6)