Official implementation of the Fake News Revealer paper
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

torch_main.py 503B

2 years ago
2 years ago
2 years ago
12345678910111213141516171819
  1. from data_loaders import build_loaders, make_dfs
  2. from learner import supervised_train
  3. from test_main import test
  4. def torch_main(config):
  5. train_df, test_df, validation_df = make_dfs(config, )
  6. train_loader = build_loaders(config, train_df, mode="train")
  7. validation_loader = build_loaders(config, validation_df, mode="validation")
  8. test_loader = build_loaders(config, test_df, mode="test")
  9. supervised_train(config, train_loader, validation_loader)
  10. test(config, test_loader)