Atsuhiko / AAMLP

0 stars 3 forks source link

第4回 機械学習のプロジェクト p73- #6

Open yuyuyuriko78 opened 3 years ago

yuyuyuriko78 commented 3 years ago

アジェンダ

p73- ファシリ:pekoさん ・if name == "main":

yuyuyuriko78 commented 3 years ago

データ前処理〜学習〜評価までで使える便利なもの

configファイル

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

シェルスクリプト run.sh

コマンドで「sh run.sh」を実行すれば3行いっきに実行できる

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"
    ),
}

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