Пример #1
0
func (jr *Jazzrobot) Load(filename string) error {
	patterns, _ := loadPatternSlice(filename)
	//chrom := []string{"C", "Db", "D", "Eb", "E", "F", "Gb", "G", "Ab", "A", "Bb", "B"}
	//chords := make(map[string][]float64)
	k := len(patterns) * 12
	for _, pattern := range patterns {
		for noteValue := 0; noteValue < 12; noteValue++ {
			weights := make([]float64, 12)
			for noteIndex, patternEntry := range pattern.Notes {
				w := pattern.Weights[noteIndex]
				newNoteValue := (noteValue + patternEntry) % 12
				weights[newNoteValue] = pattern.Weights[noteIndex] * w
			}
			id := fmt.Sprintf("%d", len(jr.chordMap))
			root := noteValue
			noteWeights := weights
			patternId := pattern.Name
			patternNotes := pattern.Notes

			jr.chordMap[id] = noteset.New(id, root, noteWeights, patternId, patternNotes)
			jr.mainModel.AddCentroid(data.New(weights, k, id))
			//chordName := fmt.Sprintf("%s %s", noteName, pattern.Name)
			//chords[chordName] = chord
		}
	}
	/*
		csvfile, err := os.Open(filename)
		if err != nil {
			return err
		}
		defer csvfile.Close()
		reader := csv.NewReader(csvfile)
		rawCSVdata, err := reader.ReadAll()
		if err != nil {
			return err
		}
		k := len(rawCSVdata)
		for _, v := range rawCSVdata {
			attributes := make([]float64, 0)
			var tag string
			for i := range v {
				if i == 0 {
					tag = v[i]
					continue
				}
				attr, _ := strconv.ParseFloat(v[i], 64)
				attributes = append(attributes, attr)
			}
			jr.mainModel.AddCentroid(data.New(attributes, k, tag))
		}*/
	jr.mainModel.UpdateCentroids()
	log.Printf("Loaded [%d, %d] number of chords\n", len(jr.chordMap), len(jr.mainModel.Centroids))
	return nil
}
Пример #2
0
func (model *Model) Classify(attributes []float64) (*data.Data, error) {
	dataItem := data.New(attributes, len(model.Centroids), "")
	dataItem.UpdateClassification(model.Centroids)
	return dataItem, nil
}