from transformers import (
    AutoTokenizer,
)
from transformers import pipeline

from depression_classification import (
    DepressionDetectionConfig,
    DepressionDetectionModel,
    DepressionPipeline
)
directory = "../depression-classification"
tokenizer = AutoTokenizer.from_pretrained(directory)

config = DepressionDetectionConfig.from_pretrained(
    directory,
)
model = DepressionDetectionModel.from_pretrained(
    pretrained_model_name_or_path=directory,
    config=config,
)

pipe = pipeline("text-classification",
                model=model,
                tokenizer=tokenizer,
                pipeline_class=DepressionPipeline,
                )

result = pipe("I don't feel hopeful.")
print(result)

Downstream Use [Optional]

Out-of-Scope Use

Any real-life, production use case is out-of-scope for this model. It is a toy project.

Bias, Risks, and Limitations

Significant research has explored bias and fairness issues with language models (see, e.g., Sheng et al. (2021) and Bender et al. (2021)). Predictions generated by the model may include disturbing and harmful stereotypes across protected classes; identity characteristics; and sensitive, social, and occupational groups.

Recommendations

I do not recommend anyone use this model.

Training Details

Training Data

Dataset originally from Depression: Reddit Dataset.

Training Procedure

Preprocessing

See preprocessing script.

Speeds, Sizes, Times

Training for 1 epoch takes ~20 minutes on local machine.

Evaluation

Testing Data, Factors & Metrics

Testing Data

More information needed

Factors

More information needed

Metrics

More information needed

Results

More information needed

jambran/depression-classification

Model Card for depression-classification

Model Details

Model Description

Uses

Direct Use

Downstream Use [Optional]

Out-of-Scope Use

Bias, Risks, and Limitations

Recommendations

Training Details

Training Data

Training Procedure

Preprocessing

Speeds, Sizes, Times

Evaluation

Testing Data, Factors & Metrics

Testing Data

Factors

Metrics

Results

jambran/
depression-classification