PythonのArgumentParserを用いた複数個の引数について、WandbのSweepを使って探索する方法が分かりません。

Question

問題の概要

PythonのArgumentParserを用いた複数個の引数を持つようなプログラムを持っています。
この時の引数をWandbのSweepを使って探索する方法が分かりません。

プログラム例

Pythonで以下のようにArgumentParserで複数個の引数を持つようなプログラムを書いています。

# example.py
import wandb, argparse

parser = argparse.ArgumentParser()
parser.add_argument('--numbers', nargs=3, type=float)

args = parser.parse_args()
loss = sum(args.numbers)
wandb.init().log({"loss": loss})

このプログラムは以下のように実行できます。

$ python example.py --numbers 1 2 3

しかし、これをWandBのSweepを用いて、numbersのそれぞれの要素を探索する方法が分かりません。
以下のような.yamlは試しました。

program: example.py
method: bayes
metric:
  name: loss
  goal: minimize
parameters:
  numbers[0]:
    min: 1
    max: 9
  numbers[1]:
    min: 1
    max: 9
  numbers[2]:
    min: 1
    max: 9

しかし、引数の名前がそのまま解釈され、ArgumentParserでパースできませんでした。
つまりこれは、以下のようなコマンドラインとして実行されてしまいます。

$ python example.py --numbers[0]=1 --numbers[1]=2 --numbers[2]=3

これを解決する何かいい方法が分かりませんか？

Comments

[can110]

Weights & Biases 使っていなので検証できませんが
- .py側では「--numbers 1 2 3」→「--numbers1 1 --numbers2 2 --numbers 3」として受け取る
- .yaml側では「numbers[0]」→「numbers1」とする。
のではダメでしょうか。

[tas0g]

ご回答ありがとうございます！
その方法ではできるのですが、記述が冗長になってしまうことを嫌っています。
もっと言うと、他にこの引数の個数をしていする様な他の引数があり、
1個のソースコードで、それに全て対応したいと思って質問していました。

[can110]

以下を読む限り、異なるparameter_name毎にしか指定できないような気がします。
https://docs.wandb.ai/guides/sweeps/configuration#parameters

[tas0g]

ご回答ありがとうございます。
やはり公式では何か良い方法はなさそうですね。
ありがとうございました！

[can110]

具体的なパラメータ次第ですが、ループ処理でadd_argument＆パースすれば、それほど複雑なコードにはならないのではないかと思います。

Answer 1

以下を読む限り、異なるparameter_name毎にしか指定できないような気がします。
Structure of the Sweep Configuration - parameters

パラメータ名を分ける手法でも、ループ処理でadd_argument＆パースすればそれほど複雑なコードにはならないのではないかと思います。

import argparse

parser = argparse.ArgumentParser()

# 1個のmesパラメータ
parser.add_argument('--mes', type=str, default='')

# 3個のnumパラメータ
arg_nums = [f'num{i+1}' for i in range(3)]
for arg in arg_nums:
    parser.add_argument(f'--{arg}', type=float, default=None)

# 呼出"～.py --num1 1 --num2 2.3 --mes abc"
args = parser.parse_args()
print(args) # Namespace(mes='abc', num1=1.0, num2=2.3, num3=None)

# num値のみ取得
dArgs = vars(args)
nums = [dArgs[key] for key in arg_nums]
#nums = [n for n in nums if n is not None] # Noneを除去したければ
print(nums) # [1.0, 2.3, None]

Comments

[tas0g]

回答ありがとうございます！
この方法で私がやりたいことができるのか検討してみます。