コード例 #1
0
ファイル: loader.go プロジェクト: timtadh/sfp
func NewItemSets(config *config.Config, min, max int) (i *ItemSets, err error) {
	parents, err := config.IntsIntsMultiMap("itemsets-parents")
	if err != nil {
		return nil, err
	}
	parentCount, err := config.IntsIntMultiMap("itemsets-parent-count")
	if err != nil {
		return nil, err
	}
	children, err := config.IntsIntsMultiMap("itemsets-children")
	if err != nil {
		return nil, err
	}
	childCount, err := config.IntsIntMultiMap("itemsets-child-count")
	if err != nil {
		return nil, err
	}
	canonKids, err := config.IntsIntsMultiMap("itemsets-canon-kids")
	if err != nil {
		return nil, err
	}
	canonKidCount, err := config.IntsIntMultiMap("itemsets-canon-kid-count")
	if err != nil {
		return nil, err
	}
	embeddings, err := config.IntsIntsMultiMap("itemsets-embeddings")
	if err != nil {
		return nil, err
	}
	i = &ItemSets{
		MinItems:      min,
		MaxItems:      max,
		Parents:       parents,
		ParentCount:   parentCount,
		Children:      children,
		ChildCount:    childCount,
		CanonKids:     canonKids,
		CanonKidCount: canonKidCount,
		Embeddings:    embeddings,
		config:        config,
	}
	return i, nil
}