zeno-ml/diffusiondb
16
1U
2 <d� � @ s| d dl mZ d dlmZmZmZmZmZmZm Z e dd� �Z3eeed�dd��Z eeed�dd ��Z eeed�d4d ��Z
d S )
� )� DataFrame)�
DistillReturn� MetricReturn� ModelReturn� ZenoOptions�distill�metric�modelc C s t td�dd�}|S )N��df�opsc S s t | |j d�S )N)� model_output)r � data_columnr5 � r �I/Users/acabrera/dev-research/22-zeno/huggingface/diffusiondb/functions.pyr s zmodel_ret.<locals>.model)r r )�namer r r r � model_ret s r r6 c C s t | d j�� d�S )N�prompt)�distill_output)r �str�lenr7 r r r �length s r c C s t t| d �� �� �d�S )NZ8image_nsfw�r �r �float�dropna�meanr9 r r r �avg_image_nswf s r c C s t t| d �� �� �d�S )NZ prompt_nsfwr r r10 r r r �avg_prompt_nsfw s r N)�pandasr �zenor r r r r r r r r r r r r r r �<module> s $ 11