FilesExpand file tree

 main

/

dvc.yaml

Copy path

BlameMore file actions

BlameMore file actions

Latest commit

 

History

History

History

53 lines (48 loc) · 1.4 KB

 main

/

dvc.yaml

Top

File metadata and controls

• Code
• Blame

53 lines (48 loc) · 1.4 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

stages:

data_ingestion:

cmd: python3 src/MLOPSCNNCLASSIFIER/pipeline/stage_01_data_ingestion.py

deps:

- src/MLOPSCNNCLASSIFIER/pipeline/stage_01_data_ingestion.py

- config/config.yaml

outs:

- artifacts/data_ingestion/KIDNEY_CT_SCAN

prepare_base_model:

cmd: python3 src/MLOPSCNNCLASSIFIER/pipeline/stage_02_prepare_base_model.py

deps:

- src/MLOPSCNNCLASSIFIER/pipeline/stage_02_prepare_base_model.py

- config/config.yaml

params:

- IMAGE_SIZE

- INCLUDE_TOP

- CLASSES

- WEIGHTS

- LEARNING_RATE

outs:

- artifacts/prepare_base_model

training:

cmd: python3 src/MLOPSCNNCLASSIFIER/pipeline/stage_03_model_training.py

deps:

- src/MLOPSCNNCLASSIFIER/pipeline/stage_03_model_training.py

- config/config.yaml

- artifacts/data_ingestion/KIDNEY_CT_SCAN

- artifacts/prepare_base_model

params:

- EPOCHS

- BATCH_SIZE

- IMAGE_SIZE

- AUGMENTATION

outs:

- artifacts/training/model.h5

evaluation:

cmd: python3 src/MLOPSCNNCLASSIFIER/pipeline/stage_04_model_evaluation.py

deps:

- src/MLOPSCNNCLASSIFIER/pipeline/stage_04_model_evaluation.py

- config/config.yaml

- artifacts/training/model.h5

- artifacts/data_ingestion/KIDNEY_CT_SCAN

params:

- BATCH_SIZE

- IMAGE_SIZE

metrics:

- scores.json:

cache: false