ClickHouse/dbms/tests/performance/linear_regression.xml

33 lines
1.5 KiB
XML
Raw Normal View History

2019-05-24 22:41:33 +00:00
<test>
<stop_conditions>
<any_of>
<total_time_ms>10000</total_time_ms>
2019-05-24 22:41:33 +00:00
</any_of>
</stop_conditions>
<preconditions>
<table_exists>test.hits</table_exists>
2020-03-03 10:09:47 +00:00
<table_exists>hits_100m_single</table_exists>
</preconditions>
2019-05-24 22:41:33 +00:00
<create_query>DROP TABLE IF EXISTS test_model</create_query>
<create_query>CREATE TABLE test_model engine = Memory as select stochasticLinearRegressionState(0.0001)(Age, Income, ParamPrice, Robotness, RefererHash) as state from test.hits</create_query>
2019-05-24 22:41:33 +00:00
2019-08-08 23:55:08 +00:00
<!-- Check model fit -->
2020-03-03 10:09:47 +00:00
<query>SELECT stochasticLinearRegressionState(0.0001, 0, 15)(Age, Income, ParamPrice, Robotness, RefererHash) FROM test.hits FORMAT Null</query>
<query>SELECT stochasticLinearRegression(Age, Income, ParamPrice, Robotness, RefererHash) FROM test.hits FORMAT Null</query>
2019-05-24 22:41:33 +00:00
2019-08-08 23:55:08 +00:00
<!-- Check model fit with Momentum -->
2020-03-03 10:09:47 +00:00
<query>SELECT stochasticLinearRegressionState(0.0001, 0, 15, 'Momentum')(Age, Income, ParamPrice, Robotness, RefererHash) FROM hits_100m_single FORMAT Null</query>
2019-05-24 22:41:33 +00:00
2019-08-08 23:55:08 +00:00
<!-- Check model fit with Nesterov -->
2020-03-03 10:09:47 +00:00
<query>SELECT stochasticLinearRegressionState(0.0001, 0, 15, 'Nesterov')(Age, Income, ParamPrice, Robotness, RefererHash) FROM hits_100m_single FORMAT Null</query>
2019-05-24 22:41:33 +00:00
2019-08-08 23:55:08 +00:00
<!-- Check model predict -->
2020-03-03 10:09:47 +00:00
<query>WITH (SELECT state FROM test_model) AS model SELECT evalMLMethod(model, Income, ParamPrice, Robotness, RefererHash) FROM hits_100m_single FORMAT Null</query>
2019-05-24 22:41:33 +00:00
<drop_query>DROP TABLE IF EXISTS test_model</drop_query>
</test>