CoolFace
Datasetpublic

guodaosun/tale-frame

TinyStories Dataset README Overview This dataset is based on TinyStories and includes structured JSON data with corresponding annotations, designed for research in controllable story generation and related tasks. Dataset Structure Each data item contains the following fields: 1. conversations Type: List Purpose: Contains the JSON of the story from: Always set to "human". value: Structured data containing entities, events, story… See the full description on the dataset page: https://huggingface.co/datasets/guodaosun/tale-frame.

sourceHugging Facemitupdated 2y agoView on Hugging Face
0likes9downloads
Dataset Card

TinyStories Dataset README

Overview

This dataset is based on TinyStories and includes structured JSON data with corresponding annotations, designed for research in controllable story generation and related tasks.

Dataset Structure

Each data item contains the following fields:

1. conversations

  • —Type: List
  • —Purpose: Contains the JSON of the story
  • —`from`: Always set to "human".
  • —`value`: Structured data containing entities, events, story structures and relationships described in JSON format.

2. chosen

  • —Type: Object
  • —Purpose: Contains the final selected story.
  • —`from`: Always set to "gpt".
  • —`value`: The accepted story text.

3. rejected

  • —Type: Object
  • —Purpose: Contains a rejected version of the story.
  • —`from`: Always set to "gpt".
  • —`value`: The rejected story text.

Structure of value Field

Units Attributes Definition

Entity

AttributeDefinitionNullability
entity_idA unique identifier for the entity, e.g., "entity_1".required
entity_nameThe name of the entity.required
entity_identityThe identity of the entity.required
entity_motivationThe motivation or goal of the entity.required
personality_traitsThe list of personality traits of the entity.required

Event

AttributeDefinitionNullability
event_idA unique identifier of the event, e.g., "event_1".required
event_timeTime the event occurs.nullable
event_locationLocation where the event happens.nullable
event_detailsA brief description for the event.required
event_importanceImportance of the event (three levels: low, medium, high).required
earlier_eventList of preceding events that caused this event, using event IDs.required
later_eventList of subsequent events resulting from this event, using event IDs.required

Relationship

AttributeDefinitionNullability
relationship_idA unique identifier for the relationship, e.g., "rel_1".required
included_entitiesThe list of entities involved in the relationship, referenced by their entity IDs.required
emotional_typeThe emotional category between the two entities in the relationship.required
action_typeThe type of behavior exhibited between the entities in the relationship.required
action_directionThe direction of the action/behavior within the relationship. "Internal", "bidirectional", "from AGENT to PATIENT".required
relationship_strengthThe strength of the relationship, categorized as "low", "medium", or "high".required
relationship_evolutionDescription of how the relationship evolves over time, such as gradually building trust.required

Story

AttributeDefinitionNullability
titleThe title of the story.required
story_descriptionA brief description of the story.required
story_structureThe structure of the story, including "beginning", "middle", "climax", and "ending".required
beginningList of event IDs that occur at the beginning of the story.required
middleList of event IDs that occur in the middle of the story.required
climaxList of event IDs that represent the climax of the story.required
endingList of event IDs that occur at the ending of the story.required

Nullability

  • —Required fields: Must contain a valid value.
  • —Nullable fields: Can be left empty.

Dataset Statistical Features

Entities

  • —Total entities: 24,154
  • —Unique entity names: 4,560
  • —Total entity traits: 4,770
  • —Entity trait distribution:
  • —Negative
  • —Neutral
  • —Positive

Stories

  • —Total stories: 9,851
  • —Emotional distribution:
  • —Negative: 1,243
  • —Neutral: 2,433
  • —Positive: 17,727

Events

  • —Total events: 68,553
  • —Emotional distribution:
  • —Negative: 9,226
  • —Neutral: 2,213
  • —Positive: 12,715

Relationships

  • —Total relationships: 69,481
  • —Emotional distribution:
  • —Negative: 12,213
  • —Neutral: 15,458
  • —Positive: 40,882

Annotations

Stories are evaluated using models like GPT-4 and annotated as "Chosen" or "Rejected" based on quality.