質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

新規登録して質問してみよう
ただいま回答率
85.35%
Python 3.x

Python 3はPythonプログラミング言語の最新バージョンであり、2008年12月3日にリリースされました。

機械学習

機械学習は、データからパターンを自動的に発見し、そこから知能的な判断を下すためのコンピューターアルゴリズムを指します。人工知能における課題のひとつです。

Q&A

0回答

1812閲覧

lightGBMにおいて、pickleロードから読み込んだ重みを使いGPUで処理を行いたい

Tera0724

総合スコア18

Python 3.x

Python 3はPythonプログラミング言語の最新バージョンであり、2008年12月3日にリリースされました。

機械学習

機械学習は、データからパターンを自動的に発見し、そこから知能的な判断を下すためのコンピューターアルゴリズムを指します。人工知能における課題のひとつです。

0グッド

0クリップ

投稿2021/01/22 07:46

現在、kaggleのコンペにおいて、lightGBMを使用して予測を行っています。学習と推論を同一のノートブックに書くとトータル時間が長くなってしまうためサブミット用のノートブックを別に作り、そこに学習済みモデルをアップロードして推論を行おうと考えプログラムを作成しました。以下がプログラムとなります。

import janestreet from tqdm.notebook import tqdm janestreet.competition.make_env.__called__ = False env = janestreet.make_env() iter_test = env.iter_test() files = glob.glob("pickelファイルが保存されているディレクトリpath") for (test_df, sample_prediction_df) in tqdm(iter_test): if (test_df.iloc[0].weight > 0): test_np = test_df.loc[:, features].values if np.isnan(test_np[:, 1:].sum()): test_np[:, 1:] = np.nan_to_num(test_np[:, 1:]) + np.isnan(test_np[:, 1:]) * f_mean test_df[features] = test_np #action = model.predict(xgb.DMatrix(test_df[features]))[0] actions = [] for file in files: with open('file', mode='rb') as f: model = pickle.load(f) actions.append(model.predict(test_df[features])[0]) action = sum(actions) / len(actions) if (action > 0.5): sample_prediction_df.action = 1 else: sample_prediction_df.action = 0 else: sample_prediction_df.action = 0 env.predict(sample_prediction_df)

janestreetを使い実際のデータをダウンロードし推論するというコンペです。
学習と推論を同時に行う場合は推論時にかかる時間は15~30分程度ですが、一度学習済みモデルを保存して別のノートブックで推論を行った場合3時間以上かかってしまいます。
おそらくGPUを使用していないと考え、lightGBMとGPUについていろいろ調べたのですが、どの資料も学習と推論を同じノートブックでやっているものばかりで解決できませんでした。
(重みをロードしたlightGBMのparamsを確認したところdvice: gpuとなってはいました。)

そのため、解決方法について教えていただきたいです。
kaggleやlightGBMを使用した経験がまだ浅いためそもそも考え方が間違っているところもあると思います。そのような指摘も是非よろしくお願いします。

気になる質問をクリップする

クリップした質問は、後からいつでもMYページで確認できます。

またクリップした質問に回答があった際、通知やメールを受け取ることができます。

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

guest

あなたの回答

tips

太字

斜体

打ち消し線

見出し

引用テキストの挿入

コードの挿入

リンクの挿入

リストの挿入

番号リストの挿入

表の挿入

水平線の挿入

プレビュー

まだ回答がついていません

会員登録して回答してみよう

アカウントをお持ちの方は

15分調べてもわからないことは
teratailで質問しよう!

ただいまの回答率
85.35%

質問をまとめることで
思考を整理して素早く解決

テンプレート機能で
簡単に質問をまとめる

質問する

関連した質問