NTK-Trainer-Demo / tasks.py
Mentors4EDU's picture
Publish ntk-trainer-demo
fca0078 verified
Raw History Blame Contribute Delete
4.01 kB
COUNTRY_CAPITALS = [
("France", "Paris"),
("Japan", "Tokyo"),
("Egypt", "Cairo"),
("Italy", "Rome"),
("Spain", "Madrid"),
("Germany", "Berlin"),
("Kenya", "Nairobi"),
("Peru", "Lima"),
("Canada", "Ottawa"),
("Norway", "Oslo"),
("Greece", "Athens"),
("Turkey", "Ankara"),
("Chile", "Santiago"),
("Cuba", "Havana"),
("Iran", "Tehran"),
("Iraq", "Baghdad"),
("Ireland", "Dublin"),
("Portugal", "Lisbon"),
("Austria", "Vienna"),
("Hungary", "Budapest"),
("Poland", "Warsaw"),
("Sweden", "Stockholm"),
("Finland", "Helsinki"),
("Denmark", "Copenhagen"),
("Thailand", "Bangkok"),
("Vietnam", "Hanoi"),
("Nepal", "Kathmandu"),
("Jordan", "Amman"),
("Lebanon", "Beirut"),
("Morocco", "Rabat"),
("Ghana", "Accra"),
("Senegal", "Dakar"),
("Ethiopia", "Addis Ababa"),
("Uganda", "Kampala"),
("Zambia", "Lusaka"),
("Argentina", "Buenos Aires"),
("Colombia", "Bogota"),
("Venezuela", "Caracas"),
("Ecuador", "Quito"),
("Uruguay", "Montevideo"),
]
ANTONYMS = [
("hot", "cold"),
("big", "small"),
("fast", "slow"),
("up", "down"),
("open", "closed"),
("light", "dark"),
("early", "late"),
("rich", "poor"),
("strong", "weak"),
("full", "empty"),
("wet", "dry"),
("hard", "soft"),
("high", "low"),
("long", "short"),
("loud", "quiet"),
("happy", "sad"),
("young", "old"),
("clean", "dirty"),
("thick", "thin"),
("near", "far"),
("sweet", "sour"),
("heavy", "light"),
("wide", "narrow"),
("deep", "shallow"),
("sharp", "dull"),
("brave", "cowardly"),
("cheap", "expensive"),
("smooth", "rough"),
("tight", "loose"),
("true", "false"),
("begin", "end"),
("buy", "sell"),
("push", "pull"),
("win", "lose"),
("give", "take"),
("love", "hate"),
("accept", "refuse"),
("arrive", "depart"),
("remember", "forget"),
("laugh", "cry"),
]
UPPERCASE_WORDS = [
"river",
"mountain",
"library",
"pyramid",
"desert",
"harbour",
"village",
"castle",
"garden",
"forest",
"bridge",
"island",
"temple",
"market",
"palace",
"valley",
"canyon",
"glacier",
"meadow",
"orchard",
"lantern",
"compass",
"anchor",
"ladder",
"window",
"mirror",
"candle",
"basket",
"pillow",
"blanket",
"piano",
"violin",
"trumpet",
"guitar",
"drum",
"flute",
"harp",
"cello",
"banjo",
"organ",
]
def _pairs(prompt_fmt: str, comp_fmt: str, items) -> str:
return "\n".join(f"{prompt_fmt.format(a)}\t{comp_fmt.format(b)}" for a, b in items)
TASKS = {
"Country to capital": _pairs("Country: {}\nCapital:", "{}", COUNTRY_CAPITALS),
"Antonyms": _pairs("The opposite of {} is", "{}", ANTONYMS),
"Uppercase the word": _pairs(
"Input: {}\nOutput:", "{}", [(w, w.upper()) for w in UPPERCASE_WORDS]
),
"Two-digit addition": "\n".join(
f"Question: {a} + {b} = ?\tAnswer: {a+b}"
for a, b in [
(14, 27),
(36, 18),
(47, 36),
(19, 8),
(52, 29),
(63, 17),
(25, 44),
(71, 9),
(38, 23),
(15, 55),
(84, 12),
(29, 31),
(46, 33),
(58, 21),
(67, 14),
(23, 49),
(31, 42),
(76, 13),
(18, 64),
(55, 27),
(42, 39),
(61, 28),
(37, 45),
(24, 57),
(69, 22),
(16, 73),
(48, 35),
(53, 19),
(32, 66),
(77, 11),
]
),
}