wesleyacheng's picture
Update README.md
61c4d1b
|
raw
history blame
1.69 kB
metadata
license: apache-2.0
datasets:
  - ag_news
language:
  - en
metrics:
  - accuracy
  - f1
pipeline_tag: text-classification
widget:
  - text: >-
      Ukraine's President Volodymyr Zelensky appears to have confirmed that his
      country's long-awaited counter-offensive against Russia has started.
      "Counter-offensive and defensive actions are taking place," he said.
    example_title: World News Example
  - text: >-
      The 2023 NFL draft class saw a record 12 quarterbacks selected in the
      first five rounds, including three inside the top five picks. The 2024
      class? Expect some similarities.
    example_title: Sports News Example
  - text: >-
      Friday's jobs report does little to change the Federal Reserve's debate
      over holding rates steady this month. But it strengthens the likelihood
      that officials would pair any June pause with a stronger preference to
      raise rates later this year.
    example_title: Business News Example
  - text: >-
      According to one recent estimate, a crew of six would require an estimated
      10,000 kilograms of food for the trip. NASA—which plans to send people to
      Mars within 2 decades—could stuff a spacecraft with prepackaged meals and
      launch additional supplies to the Red Planet in advance for the voyage
      home. But even that wouldn’t completely solve the problem.
    example_title: Science News Example

BREAKING: We are making a news topic classifier!

First posted on my Kaggle.

Here I used transfer learning on BERT with the AG News Dataset to create a multi-class news topic classifier.