Skip to content

zenn-ai/zen

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

image

Contents

Motivation

imageimage

We built ZenAI to offer Sam and others in her shoes better access to mental health support.

Overview of ZenAI

image

Data

image

Data and code for preparing the data code be found here.

Model training

image

Model training code can be found here.

Model inference

image

Model inference code can be found here.

Model testing

image

Overall, our initial testing has been focused on the following:

  • Key quantitative metrics: Our model outperforms Llama 2 on key LLM metrics.
  • Intent detection evaluation: We tested our intent-detection classifier against 30 labeled prompts. The results are shown below and highlight that our model performs quite well. Supporting code can be found here.
  • Qualitative testing: We conducted extensive user testing of the chatbot. Snippets of Zen's responses are shown above and in our UX demo. We plan to roll out our testing to other target users and therapeutic experts.

User experience

image

The UI/UX code can be found here.

A video demonstration of our UX can be found here.

Key risks & future work

Below is the summary of the key risks of ZenAI and enhancements that we'd like to consider for the future.

image

Meet our team