Create a basic example on the 'titanic' dataset

google / nitroml

NitroML is a modular, portable, and scalable model-quality benchmarking framework for Machine Learning and Automated Machine Learning (AutoML) pipelines.

Apache License 2.0

42 stars 6 forks source link

Create a basic example on the 'titanic' dataset #15

Closed cweill closed 4 years ago

cweill commented 4 years ago

Our first canonical example under https://github.com/googleinterns/nitroml/examples:

Benchmark should live in: nitroml/examples/titanic_benchmark.py
Uses the TFDSDataset to produce the titanic examples.
Uses AutoTransform to produce feature columns.
Includes a GenericTrainer using code defined in an nitroml/examples/auto_trainer.py (can be taken from the starter-project.