Exemple #1
0
    def __call__(self,
                 labels=(2, 3),
                 filename='trajectory_plots.pdf',
                 exclude_labels=(0, )):

        cmap = unsupervised_cmap(self._nclusters)
        try:
            pdf = PdfPages(join(self._odir, filename))
            for title_, tracks in self._data.iteritems():
                title = '%s (%s)' % (self._position, title_.lower())

                # checking for binary matrix
                if np.unique(tracks).size == 2:
                    is_binary = True
                    labels_ = (1, )
                else:
                    is_binary = False
                    labels_ = labels

                # trajectory plots
                fig = plots.trajectories(tracks,
                                         labels_,
                                         title=title,
                                         cmap=cmap)
                pdf.savefig(fig)

                delchars = NameValidator.defaultdeletechars.copy()
                delchars.remove('-')
                validator = NameValidator(case_sensitive='lower',
                                          deletechars=delchars)
                fname = validator.validate(title)[0] + '.csv'
                np.savetxt(join(self._odir, fname),
                           tracks,
                           fmt='%d',
                           delimiter=',')

                # boxplots
                if not is_binary:
                    ylabel = "dwell time (%s)" % self.timeunit
                    xlabel = "class labels"

                    dwell_times = self.dwell_times(tracks)
                    fig1 = plots.dwell_boxplot(dwell_times,
                                               title,
                                               ylabel=ylabel,
                                               xlabel=xlabel,
                                               exclude_labels=exclude_labels,
                                               cmap=cmap)
                    fig2 = plots.barplot(dwell_times,
                                         title,
                                         xlabel=xlabel,
                                         ylabel=ylabel,
                                         exclude_labels=exclude_labels,
                                         cmap=cmap)
                    pdf.savefig(fig1)
                    pdf.savefig(fig2)
        finally:
            pdf.close()
Exemple #2
0
    def __call__(self, labels=(2,3), filename='trajectory_plots.pdf',
                 exclude_labels=(0, )):

        cmap = unsupervised_cmap(self._nclusters)
        try:
            pdf = PdfPages(join(self._odir, filename))
            for title_, tracks in self._data.iteritems():
                title = '%s (%s)' %(self._position, title_.lower())

                # checking for binary matrix
                if np.unique(tracks).size == 2:
                    is_binary = True
                    labels_ = (1, )
                else:
                    is_binary = False
                    labels_ = labels

                # trajectory plots
                fig = plots.trajectories(tracks, labels_, title=title, cmap=cmap)
                pdf.savefig(fig)

                delchars = NameValidator.defaultdeletechars.copy()
                delchars.remove('-')
                validator = NameValidator(case_sensitive='lower',
                                          deletechars=delchars)
                fname = validator.validate(title)[0]+'.csv'
                np.savetxt(join(self._odir, fname), tracks, fmt='%d',
                           delimiter=',')

                # boxplots
                if not is_binary:
                    ylabel = "dwell time (%s)" %self.timeunit
                    xlabel = "class labels"

                    dwell_times = self.dwell_times(tracks)
                    fig1 = plots.dwell_boxplot(dwell_times, title,
                                               ylabel=ylabel, xlabel=xlabel,
                                               exclude_labels=exclude_labels,
                                               cmap=cmap)
                    fig2 = plots.barplot(dwell_times, title, xlabel=xlabel,
                                         ylabel=ylabel,
                                         exclude_labels=exclude_labels,
                                         cmap=cmap)
                    pdf.savefig(fig1)
                    pdf.savefig(fig2)
        finally:
            pdf.close()
Exemple #3
0
 def test_excludelist(self):
     "Test excludelist"
     names = ['dates', 'data', 'Other Data', 'mask']
     validator = NameValidator(excludelist=['dates', 'data', 'mask'])
     test = validator.validate(names)
     assert_equal(test, ['dates_', 'data_', 'Other_Data', 'mask_'])
 def test_excludelist(self):
     "Test excludelist"
     names = ["dates", "data", "Other Data", "mask"]
     validator = NameValidator(excludelist=["dates", "data", "mask"])
     test = validator.validate(names)
     assert_equal(test, ["dates_", "data_", "Other_Data", "mask_"])
Exemple #5
0
 def test_excludelist(self):
     "Test excludelist"
     names = ['dates', 'data', 'Other Data', 'mask']
     validator = NameValidator(excludelist=['dates', 'data', 'mask'])
     test = validator.validate(names)
     assert_equal(test, ['dates_', 'data_', 'Other_Data', 'mask_'])
Exemple #6
0
 def test_excludelist(self):
     "Test excludelist"
     names = ["dates", "data", "Other Data", "mask"]
     validator = NameValidator(excludelist=["dates", "data", "mask"])
     test = validator.validate(names)
     assert_equal(test, ["dates_", "data_", "Other_Data", "mask_"])