beimingwu
/
learnware
Not watched
1
0
Fork 0
Code
Releases 1 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: 69bc1cd41b
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '69bc1cd41b'
${ noResults }
learnware/tests/test_workflow/hetero_config.py

101 lines
6.4 kB
Raw Blame History 

  1. input_shape_list = [20, 30]  # 20-input shape of example learnware 0, 30-input shape of example learnware 1

  2. 

  3. input_description_list = [

  4.     {

  5.         "Dimension": 20,

  6.         "Description": {  # medical description

  7.             "0": "baseline value: Baseline Fetal Heart Rate (FHR)",

  8.             "1": "accelerations: Number of accelerations per second",

  9.             "2": "fetal_movement: Number of fetal movements per second",

  10.             "3": "uterine_contractions: Number of uterine contractions per second",

  11.             "4": "light_decelerations: Number of LDs per second",

  12.             "5": "severe_decelerations: Number of SDs per second",

  13.             "6": "prolongued_decelerations: Number of PDs per second",

  14.             "7": "abnormal_short_term_variability: Percentage of time with abnormal short term variability",

  15.             "8": "mean_value_of_short_term_variability: Mean value of short term variability",

  16.             "9": "percentage_of_time_with_abnormal_long_term_variability: Percentage of time with abnormal long term variability",

  17.             "10": "mean_value_of_long_term_variability: Mean value of long term variability",

  18.             "11": "histogram_width: Width of the histogram made using all values from a record",

  19.             "12": "histogram_min: Histogram minimum value",

  20.             "13": "histogram_max: Histogram maximum value",

  21.             "14": "histogram_number_of_peaks: Number of peaks in the exam histogram",

  22.             "15": "histogram_number_of_zeroes: Number of zeroes in the exam histogram",

  23.             "16": "histogram_mode: Hist mode",

  24.             "17": "histogram_mean: Hist mean",

  25.             "18": "histogram_median: Hist Median",

  26.             "19": "histogram_variance: Hist variance",

  27.         },

  28.     },

  29.     {

  30.         "Dimension": 30,

  31.         "Description": {  # business description

  32.             "0": "This is a consecutive month number, used for convenience. For example, January 2013 is 0, February 2013 is 1,..., October 2015 is 33.",

  33.             "1": "This is the unique identifier for each shop.",

  34.             "2": "This is the unique identifier for each item.",

  35.             "3": "This is the code representing the city where the shop is located.",

  36.             "4": "This is the unique identifier for the category of the item.",

  37.             "5": "This is the code representing the type of the item.",

  38.             "6": "This is the code representing the subtype of the item.",

  39.             "7": "This is the number of this type of item sold in the shop one month ago.",

  40.             "8": "This is the number of this type of item sold in the shop two months ago.",

  41.             "9": "This is the number of this type of item sold in the shop three months ago.",

  42.             "10": "This is the number of this type of item sold in the shop six months ago.",

  43.             "11": "This is the number of this type of item sold in the shop twelve months ago.",

  44.             "12": "This is the average count of items sold one month ago.",

  45.             "13": "This is the average count of this type of item sold one month ago.",

  46.             "14": "This is the average count of this type of item sold two months ago.",

  47.             "15": "This is the average count of this type of item sold three months ago.",

  48.             "16": "This is the average count of this type of item sold six months ago.",

  49.             "17": "This is the average count of this type of item sold twelve months ago.",

  50.             "18": "This is the average count of items sold in the shop one month ago.",

  51.             "19": "This is the average count of items sold in the shop two months ago.",

  52.             "20": "This is the average count of items sold in the shop three months ago.",

  53.             "21": "This is the average count of items sold in the shop six months ago.",

  54.             "22": "This is the average count of items sold in the shop twelve months ago.",

  55.             "23": "This is the average count of items in the same category sold one month ago.",

  56.             "24": "This is the average count of items in the same category sold in the shop one month ago.",

  57.             "25": "This is the average count of items of the same type sold in the shop one month ago.",

  58.             "26": "This is the average count of items of the same subtype sold in the shop one month ago.",

  59.             "27": "This is the average count of items sold in the same city one month ago.",

  60.             "28": "This is the average count of this type of item sold in the same city one month ago.",

  61.             "29": "This is the average count of items of the same type sold one month ago.",

  62.         },

  63.     },

  64. ]

  65. 

  66. output_description_list = [

  67.     {

  68.         "Dimension": 1,

  69.         "Description": {"0": "length of stay: Length of hospital stay (days)"},  # medical description

  70.     },

  71.     {

  72.         "Dimension": 1,

  73.         "Description": {  # business description

  74.             "0": "sales of the item in the next day: Number of items sold in the next day"

  75.         },

  76.     },

  77. ]

  78. 

  79. user_description_list = [

  80.     {

  81.         "Dimension": 15,

  82.         "Description": {  # medical description

  83.             "0": "Whether the patient is on thyroxine medication (0: No, 1: Yes)",

  84.             "1": "Whether the patient has been queried about thyroxine medication (0: No, 1: Yes)",

  85.             "2": "Whether the patient is on antithyroid medication (0: No, 1: Yes)",

  86.             "3": "Whether the patient has undergone thyroid surgery (0: No, 1: Yes)",

  87.             "4": "Whether the patient has been queried about hypothyroidism (0: No, 1: Yes)",

  88.             "5": "Whether the patient has been queried about hyperthyroidism (0: No, 1: Yes)",

  89.             "6": "Whether the patient is pregnant (0: No, 1: Yes)",

  90.             "7": "Whether the patient is sick (0: No, 1: Yes)",

  91.             "8": "Whether the patient has a tumor (0: No, 1: Yes)",

  92.             "9": "Whether the patient is taking lithium (0: No, 1: Yes)",

  93.             "10": "Whether the patient has a goitre (enlarged thyroid gland) (0: No, 1: Yes)",

  94.             "11": "Whether TSH (Thyroid Stimulating Hormone) level has been measured (0: No, 1: Yes)",

  95.             "12": "Whether T3 (Triiodothyronine) level has been measured (0: No, 1: Yes)",

  96.             "13": "Whether TT4 (Total Thyroxine) level has been measured (0: No, 1: Yes)",

  97.             "14": "Whether T4U (Thyroxine Utilization) level has been measured (0: No, 1: Yes)",

  98.         },

  99.     }

  100. ]