Open yuyuyuriko78 opened 3 years ago
if __name__ == "__main__":
run(fold=0)
run(fold=1)
run(fold=2)
↓ argparseを使うと
import argparse
if name == "main": parser = argparse.ArgumentParser() parser.add_argument( "--fold", type=int ) args = parser.parse_args() run(fold=args.fold)
- コマンド
python train.py --fold 2
python trrain.py --fold 0 python trrain.py --fold 1 python trrain.py --fold 2
### ■モデルで使ってみる
- model_dispacher.py
```python
from sklearn import tree
models = {
"decision_tree_gini": tree.DecisionTreeClassifier(
criterion="gini"
),
"decision_tree_entropy": tree.DecisionTreeClassifier(
criterion="entropy"
),
}
import model_dispacher
...
# モデルを選んで学習させる部分
clf = model_dispacher.models[model]
...
if name == "main": parser = argparse.ArgumentParser()
parser.add_argument(
"--fold",
type=int
)
parser.add_argument(
"--model",
type=str
)
args = parser.parse_args()
run(
fold=args.fold,
model=args.model
)
- コマンド
python train.py --fold 2 --model decision_tree_gini
アジェンダ
p73- ファシリ:pekoさん ・if name == "main":