salesforce / TransmogrifAI

TransmogrifAI (pronounced trăns-mŏgˈrə-fī) is an AutoML library for building modular, reusable, strongly typed machine learning workflows on Apache Spark with minimal hand-tuning
https://transmogrif.ai
BSD 3-Clause "New" or "Revised" License
2.24k stars 392 forks source link

Train a XGBoost model and use it to score to test MLeap compatability #489

Closed TuanNguyen27 closed 4 years ago

TuanNguyen27 commented 4 years ago

Refactor & add a test in OpWorkflowModelLocalTest.scala to make sure that we can score new records with a XGBoost classifier.

codecov[bot] commented 4 years ago

Codecov Report

Merging #489 into master will increase coverage by 5.60%. The diff coverage is 100.00%.

Impacted file tree graph

@@            Coverage Diff             @@
##           master     #489      +/-   ##
==========================================
+ Coverage   81.40%   87.00%   +5.60%     
==========================================
  Files         345      345              
  Lines       11684    11697      +13     
  Branches      379      388       +9     
==========================================
+ Hits         9511    10177     +666     
+ Misses       2173     1520     -653     
Impacted Files Coverage Δ
...sification/BinaryClassificationModelSelector.scala 98.41% <100.00%> (+0.16%) :arrow_up:
...p/stages/impl/selector/DefaultSelectorParams.scala 100.00% <100.00%> (ø)
...s/impl/preparators/DerivedFeatureFilterUtils.scala 93.08% <0.00%> (+0.62%) :arrow_up:
...sforce/op/stages/impl/feature/Transmogrifier.scala 98.05% <0.00%> (+0.83%) :arrow_up:
...com/salesforce/op/features/FeatureSparkTypes.scala 99.14% <0.00%> (+0.85%) :arrow_up:
...esforce/op/stages/impl/feature/TextTokenizer.scala 97.36% <0.00%> (+1.31%) :arrow_up:
...a/com/salesforce/op/readers/JoinedDataReader.scala 94.23% <0.00%> (+1.92%) :arrow_up:
...op/stages/impl/selector/ModelSelectorSummary.scala 91.83% <0.00%> (+2.04%) :arrow_up:
...p/stages/impl/feature/SmartTextMapVectorizer.scala 100.00% <0.00%> (+2.12%) :arrow_up:
...n/scala/com/salesforce/op/dsl/RichMapFeature.scala 67.64% <0.00%> (+2.94%) :arrow_up:
... and 47 more

Continue to review full report at Codecov.

Legend - Click here to learn more Δ = absolute <relative> (impact), ø = not affected, ? = missing data Powered by Codecov. Last update 7baf0f8...a1f4e54. Read the comment docs.