models.yml

models:
  - name: "iris_lr"
    description: "Iris dataset for logistic regression"
    model_path: "iris_lr/iris_lr.pkl"
    data_schema_path: "iris_lr/iris_lr.json"
    target: "target"

  - name: "toxic_lr"
    description: "Toxic comments classification with TF/IDF and LogisticRegression"
    model_path: "toxic_lr/toxic_lr.pkl"
    data_schema_path: "toxic_lr/toxic_lr.json"
    target: ["identity_hate", "insult", "obscene", "severe_toxic", "threat", "toxic"]

  - name: "instagram_rf"
    description: "Classification fake user on instagram using just basic profile information"
    model_path: "instagram/instgram_rf.pkl"
    data_schema_path: "instagram/instagram.json"
    target: ["rating"]

  - name: "toxic_lstm"
    description: "Toxic comments classification with Keras LSTM"
    model_path: "toxic_lstm/keras.pkl"
    data_schema_path: "toxic_lstm/toxic_lstm.json"
    target: ["identity_hate", "insult", "obscene", "severe_toxic", "threat", "toxic"]