config_example.yaml

config:
  data:
    input_data:
      clear: true
      movielens:
        use: true
        type: ml-1m
        path: movielens
        transformations: rec_tool.transformations.movielens.prepare_data
      goodreads:
        use: false
        type: goodreads
        transformations: rec_tool.transformations.goodreads.get_goodreads_data
      bookcrossing:
        use: false
        type: bookcrossing
        transformations: rec_tool.transformations.bookcrossing.bookcrossing_converting
      behance:
        use: false
        type: behance
        transformations: rec_tool.transformations.behance.behance_converting

  model:
    model: [
      rec_tool.models.svd.svd,
    ]
    loss: rec_tool.losses.mean_squared_error
    metrics: [
      rec_tool.metrics.accuracy,
      rec_tool.metrics.rmse,
      rec_tool.metrics.mae
    ]
    batch_size: [1024, 2048, 4096]
    epoch: [50, 100, 200]
    optimizers: adam
    grid_search: True
    learning_rate: 0.01


  result:
    model: /tmp/rec_tool/model/
    log: /tmp/rec_tool/log/
    results_csv: run_results.csv
    log_to_ml_flow: false
    clear: true