transform

old TransE-like models
git clone https://esimon.eu/repos/transform.git
Log | Files | Refs | README

test.py (1125B)


      1 #!/usr/bin/env python2
      2 
      3 from __future__ import print_function
      4 import sys
      5 
      6 from dataset import *
      7 from model import *
      8 from meta_model import *
      9 from config import *
     10 
     11 if __name__ == '__main__':
     12     if len(sys.argv)<3:
     13         print('Usage: {0} data config [models]...'.format(sys.argv[0]), file=sys.stderr)
     14         sys.exit(1)
     15     data = sys.argv[1]
     16     config_path = sys.argv[2]
     17 
     18     if len(sys.argv)<4: model_pathes = None
     19     elif len(sys.argv)>4: model_pathes = sys.argv[3:]
     20     else: model_pathes = sys.argv[3]
     21 
     22     config = load_config(config_path)
     23     if model_pathes is None:
     24         if not config.get('meta', False):
     25             model_pathes = '{0}/{1}.best'.format(config['best model save path'], config['model name'])
     26     if not config.get('meta', False) and isinstance(model_pathes, list):
     27         print('Error: multiple model specified while running in single mode', file=sys.stderr)
     28         sys.exit(1)
     29     ModelType = Meta_model if config.get('meta', False) else Model
     30 
     31     data = Dataset(data, config)
     32     model = ModelType(data, config, model_pathes)
     33     model.build_test()
     34     model.test_all()
     35     model.test()