├── .github └── workflows │ └── python-publish.yml ├── .gitignore ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── deployment ├── cloud_function │ ├── main.py │ └── requirements.txt ├── cloud_run │ ├── .dockerignore │ ├── Dockerfile │ ├── cloudbuild.yaml │ ├── main.py │ └── requirements.txt ├── deploy.sh ├── deployment_instructions.pdf ├── requirements.txt ├── setup │ ├── columns_metadata.py │ ├── environmental_variables.py │ └── hyperparameters.py ├── undeploy.sh └── vertex_pipeline │ ├── .dockerignore │ ├── Dockerfile │ ├── custom_code │ └── custom_reward.py │ ├── requirements.txt │ └── vertex_pipeline.py ├── images ├── proposed_gcp_implementation_architecture.png └── step_by_step_optimus.png ├── optimus ├── __init__.py ├── actions_lib │ ├── __init__.py │ ├── actions.py │ └── base_actions.py ├── agent_lib │ ├── __init__.py │ ├── agents.py │ ├── base_agent.py │ └── tabnet.py ├── data_pipeline_lib │ ├── __init__.py │ ├── base_data_pipeline.py │ └── data_pipelines.py ├── preprocessing_lib │ ├── __init__.py │ └── base_preprocessing.py ├── reward_lib │ ├── __init__.py │ ├── base_reward.py │ └── rewards.py ├── trainer_lib │ ├── __init__.py │ ├── base_trainer.py │ ├── losses.py │ ├── optimizers.py │ ├── schedules.py │ └── trainers.py └── training_setup.py ├── requirements.txt ├── setup.py └── tests ├── actions_test.py ├── agents_test.py ├── base_actions_test.py ├── base_agent_test.py ├── base_data_pipeline_test.py ├── base_preprocessing_test.py ├── base_reward_test.py ├── base_trainer_test.py ├── data_pipelines_test.py ├── losses_test.py ├── optimizers_test.py ├── rewards_test.py ├── schedules_test.py ├── tabnet_test.py ├── trainers_test.py └── training_setup_test.py /.github/workflows/python-publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/.github/workflows/python-publish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/.gitignore -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/README.md -------------------------------------------------------------------------------- /deployment/cloud_function/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/deployment/cloud_function/main.py -------------------------------------------------------------------------------- /deployment/cloud_function/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/deployment/cloud_function/requirements.txt -------------------------------------------------------------------------------- /deployment/cloud_run/.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/deployment/cloud_run/.dockerignore -------------------------------------------------------------------------------- /deployment/cloud_run/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/deployment/cloud_run/Dockerfile -------------------------------------------------------------------------------- /deployment/cloud_run/cloudbuild.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/deployment/cloud_run/cloudbuild.yaml -------------------------------------------------------------------------------- /deployment/cloud_run/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/deployment/cloud_run/main.py -------------------------------------------------------------------------------- /deployment/cloud_run/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/deployment/cloud_run/requirements.txt -------------------------------------------------------------------------------- /deployment/deploy.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/deployment/deploy.sh -------------------------------------------------------------------------------- /deployment/deployment_instructions.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/deployment/deployment_instructions.pdf -------------------------------------------------------------------------------- /deployment/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/deployment/requirements.txt -------------------------------------------------------------------------------- /deployment/setup/columns_metadata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/deployment/setup/columns_metadata.py -------------------------------------------------------------------------------- /deployment/setup/environmental_variables.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/deployment/setup/environmental_variables.py -------------------------------------------------------------------------------- /deployment/setup/hyperparameters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/deployment/setup/hyperparameters.py -------------------------------------------------------------------------------- /deployment/undeploy.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/deployment/undeploy.sh -------------------------------------------------------------------------------- /deployment/vertex_pipeline/.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/deployment/vertex_pipeline/.dockerignore -------------------------------------------------------------------------------- /deployment/vertex_pipeline/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/deployment/vertex_pipeline/Dockerfile -------------------------------------------------------------------------------- /deployment/vertex_pipeline/custom_code/custom_reward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/deployment/vertex_pipeline/custom_code/custom_reward.py -------------------------------------------------------------------------------- /deployment/vertex_pipeline/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/deployment/vertex_pipeline/requirements.txt -------------------------------------------------------------------------------- /deployment/vertex_pipeline/vertex_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/deployment/vertex_pipeline/vertex_pipeline.py -------------------------------------------------------------------------------- /images/proposed_gcp_implementation_architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/images/proposed_gcp_implementation_architecture.png -------------------------------------------------------------------------------- /images/step_by_step_optimus.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/images/step_by_step_optimus.png -------------------------------------------------------------------------------- /optimus/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/optimus/__init__.py -------------------------------------------------------------------------------- /optimus/actions_lib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/optimus/actions_lib/__init__.py -------------------------------------------------------------------------------- /optimus/actions_lib/actions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/optimus/actions_lib/actions.py -------------------------------------------------------------------------------- /optimus/actions_lib/base_actions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/optimus/actions_lib/base_actions.py -------------------------------------------------------------------------------- /optimus/agent_lib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/optimus/agent_lib/__init__.py -------------------------------------------------------------------------------- /optimus/agent_lib/agents.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/optimus/agent_lib/agents.py -------------------------------------------------------------------------------- /optimus/agent_lib/base_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/optimus/agent_lib/base_agent.py -------------------------------------------------------------------------------- /optimus/agent_lib/tabnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/optimus/agent_lib/tabnet.py -------------------------------------------------------------------------------- /optimus/data_pipeline_lib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/optimus/data_pipeline_lib/__init__.py -------------------------------------------------------------------------------- /optimus/data_pipeline_lib/base_data_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/optimus/data_pipeline_lib/base_data_pipeline.py -------------------------------------------------------------------------------- /optimus/data_pipeline_lib/data_pipelines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/optimus/data_pipeline_lib/data_pipelines.py -------------------------------------------------------------------------------- /optimus/preprocessing_lib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/optimus/preprocessing_lib/__init__.py -------------------------------------------------------------------------------- /optimus/preprocessing_lib/base_preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/optimus/preprocessing_lib/base_preprocessing.py -------------------------------------------------------------------------------- /optimus/reward_lib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/optimus/reward_lib/__init__.py -------------------------------------------------------------------------------- /optimus/reward_lib/base_reward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/optimus/reward_lib/base_reward.py -------------------------------------------------------------------------------- /optimus/reward_lib/rewards.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/optimus/reward_lib/rewards.py -------------------------------------------------------------------------------- /optimus/trainer_lib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/optimus/trainer_lib/__init__.py -------------------------------------------------------------------------------- /optimus/trainer_lib/base_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/optimus/trainer_lib/base_trainer.py -------------------------------------------------------------------------------- /optimus/trainer_lib/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/optimus/trainer_lib/losses.py -------------------------------------------------------------------------------- /optimus/trainer_lib/optimizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/optimus/trainer_lib/optimizers.py -------------------------------------------------------------------------------- /optimus/trainer_lib/schedules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/optimus/trainer_lib/schedules.py -------------------------------------------------------------------------------- /optimus/trainer_lib/trainers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/optimus/trainer_lib/trainers.py -------------------------------------------------------------------------------- /optimus/training_setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/optimus/training_setup.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/setup.py -------------------------------------------------------------------------------- /tests/actions_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/tests/actions_test.py -------------------------------------------------------------------------------- /tests/agents_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/tests/agents_test.py -------------------------------------------------------------------------------- /tests/base_actions_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/tests/base_actions_test.py -------------------------------------------------------------------------------- /tests/base_agent_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/tests/base_agent_test.py -------------------------------------------------------------------------------- /tests/base_data_pipeline_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/tests/base_data_pipeline_test.py -------------------------------------------------------------------------------- /tests/base_preprocessing_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/tests/base_preprocessing_test.py -------------------------------------------------------------------------------- /tests/base_reward_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/tests/base_reward_test.py -------------------------------------------------------------------------------- /tests/base_trainer_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/tests/base_trainer_test.py -------------------------------------------------------------------------------- /tests/data_pipelines_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/tests/data_pipelines_test.py -------------------------------------------------------------------------------- /tests/losses_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/tests/losses_test.py -------------------------------------------------------------------------------- /tests/optimizers_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/tests/optimizers_test.py -------------------------------------------------------------------------------- /tests/rewards_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/tests/rewards_test.py -------------------------------------------------------------------------------- /tests/schedules_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/tests/schedules_test.py -------------------------------------------------------------------------------- /tests/tabnet_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/tests/tabnet_test.py -------------------------------------------------------------------------------- /tests/trainers_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/tests/trainers_test.py -------------------------------------------------------------------------------- /tests/training_setup_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-marketing-solutions/optimus/HEAD/tests/training_setup_test.py --------------------------------------------------------------------------------