def main(): parser = argparse.ArgumentParser() parser.add_argument("inputList", #nargs="?" , help=("Training sample list."), #default=config.defaultTrainSampleList ) parser.add_argument("outputDir", nargs="?" , help="Music output directory", default = config.defaultOutputDir ) #parser.add_argument("modelName", nargs="?" , # help="model filename", # default=config.defaultModelFilename # ) args = parser.parse_args() trainSampList = sampleLoader.parseFileList(args.inputList) trainFeatsList = [] for trainSampFilename in trainSampList: trainSamp = sampleLoader.loadTrainSample(trainSampFilename) trainFeat = featureManager.extractTrainFeat(trainSamp) trainFeatsList.append(trainFeat) trainFeatFilename = config.getTrainInFeatFilename(args) featureManager.saveJson(trainFeatsList, trainFeatFilename); m = model.getModelObj(); #modelName = config.getModelFilename(args) m.train(args); #load features from config.getTrainInFeatFilename(args)
def main(): parser = argparse.ArgumentParser() parser.add_argument( "inputList", #nargs="?" , help=("Training sample list."), #default=config.defaultTrainSampleList ) parser.add_argument("outputDir", nargs="?", help="Music output directory", default=config.defaultOutputDir) #parser.add_argument("modelName", nargs="?" , # help="model filename", # default=config.defaultModelFilename # ) args = parser.parse_args() trainSampList = sampleLoader.parseFileList(args.inputList) trainFeatsList = [] for trainSampFilename in trainSampList: trainSamp = sampleLoader.loadTrainSample(trainSampFilename) trainFeat = featureManager.extractTrainFeat(trainSamp) trainFeatsList.append(trainFeat) trainFeatFilename = config.getTrainInFeatFilename(args) featureManager.saveJson(trainFeatsList, trainFeatFilename) m = model.getModelObj() #modelName = config.getModelFilename(args) m.train(args)
import config import sampleLoader import featureManager trainSampList = sampleLoader.parseFileList(config.unittestTrainSampleList) for trainSampFilename in trainSampList: print("[INFO] Testing training sample: " + trainSampFilename) trainSamp = sampleLoader.loadTrainSample(trainSampFilename) print("[INFO] Extracting samples") trainFeat = featureManager.extractTrainFeat(trainSamp) #print(trainFeat) print(trainFeat['scoreFeats']) for key, feat in trainFeat['scoreFeats'].items(): print(key) assert ( len(feat) == 4 ), 'the number of feature points doesn\'t match the test sample' for key, feat in trainFeat['perfFeats'].items(): print(key) assert ( len(feat) == 4 ), 'the number of feature points doesn\'t match the test sample' print('Tests passed')
import config import sampleLoader import featureManager trainSampList = sampleLoader.parseFileList(config.unittestTrainSampleList) for trainSampFilename in trainSampList: print("[INFO] Testing training sample: " + trainSampFilename) trainSamp = sampleLoader.loadTrainSample(trainSampFilename) print("[INFO] Extracting samples") trainFeat = featureManager.extractTrainFeat(trainSamp) #print(trainFeat) print(trainFeat['scoreFeats']) for key,feat in trainFeat['scoreFeats'].items(): print(key) assert(len(feat) == 4), 'the number of feature points doesn\'t match the test sample' for key, feat in trainFeat['perfFeats'].items(): print(key) assert(len(feat) == 4), 'the number of feature points doesn\'t match the test sample' print('Tests passed')