MLproject

entry_points:
    main:
        parameters:
            k: {type: int, default: 10}
            train_path: path
            test_path: path
            seed: {type: int, default: 1}
            weights: {type: string, default: "0.3,0.3,0.3"}
            normalization: string
            algorithm: string
            diversity: string
            metadata_path: {type: string, default: ""}
            baseline: string
            shift: float
            cache_dir: {type: string, default: ""}
            discounts: string
        command: "python3 main.py \
            --k {k} --train_path {train_path} --test_path {test_path} \
            --seed {seed} --weights {weights} --normalization {normalization} \
            --algorithm {algorithm} --diversity {diversity} \
            --metadata_path {metadata_path} --baseline {baseline} \
            --shift {shift} --cache_dir {cache_dir} \
            --discounts {discounts}"