MLproject

# Author: Lukas Heumos
# Email: lukas.heumos@posteo.net
# Version: 1.0.1
# License: MIT

name: lcep

# conda_env: environment.yml
docker_env:
  image: ghcr.io/mlf-core/lcep:1.0.1
  volumes: ["${PWD}/data:/data"]

entry_points:
  main:
    parameters:
      cuda: {type: string, default: 'True'}
      max_epochs: {type: int, default: 25}
      general-seed: {type: int, default: 0}
      xgboost-seed: {type: int, default: 0}
      single-precision-histogram: {type: string, default: 'True'}
      training-data: {type: string, default: 'train.tsv'}
      test-data: {type: string, default: 'test.tsv'}
    command: |
          python lcep/lcep.py \
            --cuda {cuda} \
            --max_epochs {max_epochs} \
            --general-seed {general-seed} \
            --xgboost-seed {xgboost-seed} \
            --single-precision-histogram {single-precision-histogram} \
            --training-data {training-data} \
            --test-data {test-data} \