BigQueryでブートストラップサンプリング
概要
やったこと
データ分析をしていると、ブートストラップサンプリングを(リンクのノンパラメトリック)したくなることがあります。
※ブートストラップ法の説明はここではしません。
PythonやRのライブラリでサンプリングできますが、BigQueryからデータを移行する必要があります。
面倒くさいのでSQLオンリー(BigQuery Scriptingを利用していますが)で処理できると嬉しいので作成しました。
注意
現状、以下の問題点があります。
- 今回のやり方はコンピューティング層とストレージ層を行き来し、終始オンメモリで実行できる訳ではないので実行時間はかなり遅いです。
- また、BigQuery Scriptingを使用しているためスキャン量の推測が見れません。データ量や試行回数が多くなるとスキャン量は指数関数的に増えます。
- 時間単位でテーブルへの書き込み制限がある場合、試行回数を大きくするとエラーになります。
PythonやRで楽に処理(BigQueryのテーブルをそのまま読み込める頭)できるなら、そちらを強くおすすめします(悲しい)。
今後の展望
オンメモリで処理する上手いやり方があるかもしれないので、模索したいと思います。
BQはJavascriptでもデータ処理できるので、そちらで実装した方が早くて安いかもしれません。
もし良いやり方がありましたらご教示くださいm(_ _)m
実験用データの作成
まずはこちらで実験用の正規分布に従った乱数をSQLで作成します
-- 乱数作成パラメータ
DECLARE population_sample_number INT64;
SET population_sample_number = 10;
-- 正規分布乱数の作成
CREATE OR REPLACE TEMP TABLE norm_sample AS (
WITH
generate_random AS (
SELECT
index,
RAND() AS x,
RAND() AS y,
FROM (SELECT GENERATE_ARRAY(1, population_sample_number, 1) AS idx)
,UNNEST(idx) AS index
)
,
generate_norm AS (
SELECT
index,
SQRT(-2*LOG(x))*COS(2*ACOS(-1)*y) AS norm
FROM generate_random
)
SELECT
*
FROM generate_norm
);
試しにサンプルサイズ10,000で乱数を生成すると、
平均は-0.0006849098015
不偏分散は1.00165907
になりました。
平均0、分散1の標準正規分布の乱数になってそうですね。
復元抽出でリサンプリング
-- 再標本化パラメータ
DECLARE seed INT64;-- 乱数生成のシード値
DECLARE trial_number INT64;-- 施行回数
DECLARE resample_number INT64;-- サンプルサイズ
DECLARE i INT64;
DECLARE seed_array ARRAY<INT64>;
SET trial_number = 10; -- 試行したい回数を入力
SET resample_number = 10; -- サンプルサイズを入力
SET seed_array = GENERATE_ARRAY(1,trial_number,1);
SET i = 0;
WHILE i < trial_number DO
SET seed = (SELECT seed_array[OFFSET(i)]);
IF i = 0 THEN
CREATE OR REPLACE TEMP TABLE resample_table AS (
SELECT
STRUCT(i AS iter_number, resampled) AS iter_table
FROM(
WITH
sample_index AS(
SELECT
ABS(MOD(FARM_FINGERPRINT(CONCAT(SAFE_CAST(index AS STRING),SAFE_CAST(seed AS STRING))),resample_number)) + 1 AS index
FROM (SELECT GENERATE_ARRAY(1, resample_number, 1) AS idx),UNNEST(idx) AS index
)
,
resample AS (
SELECT
*
FROM sample_index
JOIN norm_sample USING(index)
)
SELECT
ARRAY_AGG(STRUCT(index, norm)) AS resampled
FROM resample
)
);
ELSE
CREATE OR REPLACE TEMP TABLE resample_table AS (
SELECT * FROM resample_table
UNION ALL
SELECT
STRUCT(i AS iter_number, resampled) AS iter_table
FROM(
WITH
sample_index AS(
SELECT
ABS(MOD(FARM_FINGERPRINT(CONCAT(SAFE_CAST(index AS STRING),SAFE_CAST(seed AS STRING))),resample_number)) + 1 AS index
FROM (SELECT GENERATE_ARRAY(1, resample_number, 1) AS idx),UNNEST(idx) AS index
)
,
resample AS (
SELECT
*
FROM sample_index
JOIN norm_sample USING(index)
)
SELECT
ARRAY_AGG(STRUCT(index, norm)) AS resampled
FROM resample
)
);
END IF;
SET i = i + 1;
END WHILE;
-- 再標本化パラメータ
DECLARE seed INT64;-- 乱数生成のシード値
DECLARE trial_number INT64;-- 施行回数
DECLARE resample_number INT64;-- サンプルサイズ
DECLARE i INT64;
DECLARE seed_array ARRAY<INT64>;
SET trial_number = 10; -- 試行したい回数を入力
SET resample_number = 10; -- サンプルサイズを入力
SET seed_array = GENERATE_ARRAY(1,trial_number,1);
SET i = 0;
WHILE i < trial_number DO
SET seed = (SELECT seed_array[OFFSET(i)]);
IF i = 0 THEN
CREATE OR REPLACE TEMP TABLE resample_table AS (
SELECT
STRUCT(i AS iter_number, resampled) AS iter_table
FROM(
WITH
sample_index AS(
SELECT
ABS(MOD(FARM_FINGERPRINT(CONCAT(SAFE_CAST(index AS STRING),SAFE_CAST(seed AS STRING))),resample_number)) + 1 AS index
FROM (SELECT GENERATE_ARRAY(1, resample_number, 1) AS idx),UNNEST(idx) AS index
)
,
resample AS (
SELECT
*
FROM sample_index
JOIN norm_sample USING(index)
)
SELECT
ARRAY_AGG(STRUCT(index, norm)) AS resampled
FROM resample
)
);
ELSE
CREATE OR REPLACE TEMP TABLE resample_table AS (
SELECT * FROM resample_table
UNION ALL
SELECT
STRUCT(i AS iter_number, resampled) AS iter_table
FROM(
WITH
sample_index AS(
SELECT
ABS(MOD(FARM_FINGERPRINT(CONCAT(SAFE_CAST(index AS STRING),SAFE_CAST(seed AS STRING))),resample_number)) + 1 AS index
FROM (SELECT GENERATE_ARRAY(1, resample_number, 1) AS idx),UNNEST(idx) AS index
)
,
resample AS (
SELECT
*
FROM sample_index
JOIN norm_sample USING(index)
)
SELECT
ARRAY_AGG(STRUCT(index, norm)) AS resampled
FROM resample
)
);
END IF;
SET i = i + 1;
END WHILE;
resample_tableの中身↓
試行回数毎にARRAY型でサンプルを保持しています。
集計して結果を確認
先程作成した乱数を使って、以下のパラメータで試してみました。実行時間は約6分です。
trial_number(試行回数) = 100;
resample_number(サンプルサイズ) = 10000;
以下のクエリで集計し、結果を確認しました。
-- 集計
WITH
unnest_table AS (
SELECT DISTINCT
iter_table.iter_number,
sample.index,
sample.norm
FROM resample_table
,UNNEST(iter_table.resampled) AS sample
)
SELECT
iter_number,
AVG(norm) AS average,
VAR_POP(norm) AS var,
FROM unnest_table
GROUP BY iter_number
以下のような結果になりました。
概ね、先程作成したサンプルの特徴を反映してそうです。
Author And Source
この問題について(BigQueryでブートストラップサンプリング), 我々は、より多くの情報をここで見つけました https://qiita.com/ognek/items/c590fa34ca7a59a2e284著者帰属:元の著者の情報は、元のURLに含まれています。著作権は原作者に属する。
Content is automatically searched and collected through network algorithms . If there is a violation . Please contact us . We will adjust (correct author information ,or delete content ) as soon as possible .