Register
Login
Resources
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Product
Data Engine LLMs Platform Enterprise
Pricing Explore
Connect to our Discord channel

dvc.yaml 1.6 KB

You have to be logged in to leave a comment. Sign In
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
  1. artifacts:
  2. stackoverflow-dataset:
  3. path: data/data.xml
  4. type: dataset
  5. desc: Initial XML StackOverflow dataset (raw data)
  6. text-classification:
  7. path: model.pkl
  8. desc: Detect whether the given stackoverflow question should have R language tag
  9. type: model
  10. labels:
  11. - nlp
  12. - classification
  13. - stackoverflow
  14. stages:
  15. prepare:
  16. cmd: python src/prepare.py data/data.xml
  17. deps:
  18. - data/data.xml
  19. - src/prepare.py
  20. params:
  21. - prepare.seed
  22. - prepare.split
  23. outs:
  24. - data/prepared
  25. featurize:
  26. cmd: python src/featurization.py data/prepared data/features
  27. deps:
  28. - data/prepared
  29. - src/featurization.py
  30. params:
  31. - featurize.max_features
  32. - featurize.ngrams
  33. outs:
  34. - data/features
  35. train:
  36. cmd: python src/train.py data/features model.pkl
  37. deps:
  38. - data/features
  39. - src/train.py
  40. params:
  41. - train.min_split
  42. - train.n_est
  43. - train.seed
  44. outs:
  45. - model.pkl
  46. evaluate:
  47. cmd: python src/evaluate.py model.pkl data/features
  48. deps:
  49. - data/features
  50. - model.pkl
  51. - src/evaluate.py
  52. outs:
  53. - eval
  54. metrics:
  55. - eval/metrics.json
  56. plots:
  57. - ROC:
  58. template: simple
  59. x: fpr
  60. y:
  61. eval/plots/sklearn/roc/train.json: tpr
  62. eval/plots/sklearn/roc/test.json: tpr
  63. - Confusion-Matrix:
  64. template: confusion
  65. x: actual
  66. y:
  67. eval/plots/sklearn/cm/train.json: predicted
  68. eval/plots/sklearn/cm/test.json: predicted
  69. - Precision-Recall:
  70. template: simple
  71. x: recall
  72. y:
  73. eval/plots/sklearn/prc/train.json: precision
  74. eval/plots/sklearn/prc/test.json: precision
  75. - eval/plots/images/importance.png
Tip!

Press p or to see the previous file or, n or to see the next file

Comments

Loading...