analysis(matlab): rate + count outputs, variation batch 2
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
@@ -1,11 +1,12 @@
|
||||
% Variation learning-curve plot, in the style of the paper:
|
||||
% Variation learning-curve plots, in the style of the paper:
|
||||
% "Lines indicate mean (and SEM) across animals in the anodal (red) and
|
||||
% control (blue) groups."
|
||||
% Plots mean +/- SEM successful reaches per training day for the treatment /
|
||||
% anodal group (stim = 1, red) and the control group (stim = 0, blue), reading
|
||||
% this folder's data.csv and saving learning_curve.png. The per-group N is read
|
||||
% from the data (each variation pools different groups), so the legend shows the
|
||||
% actual counts. Training day is 1-indexed (our day 0 = the paper's "Day 1").
|
||||
% Produces TWO figures from this folder's data.csv:
|
||||
% learning_curve.png # successes (count) per training day
|
||||
% learning_curve_rate.png success rate (success/attempts) per training day
|
||||
% anodal / treatment = stim 1 (red); control = stim 0 (blue). Per-group N is
|
||||
% read from the data. Training day is 1-indexed (our day 0 = paper "Day 1") and
|
||||
% the x-axis tick labels are drawn vertically.
|
||||
% Run: matlab -batch "plotcurve"
|
||||
% (Copy of analysis/matlab/variation_plot.m; see make_variation_plot.m.)
|
||||
|
||||
@@ -14,15 +15,20 @@ if isempty(here); here = pwd; end
|
||||
vname = regexprep(here, '.*[/\\]', '');
|
||||
|
||||
D = readtable(fullfile(here, 'data.csv'), 'TextType', 'string');
|
||||
days = unique(D.day); % 0-indexed
|
||||
xd = days + 1; % plot as 1-indexed training day (paper axis)
|
||||
|
||||
days = unique(D.day);
|
||||
xd = days + 1; % plot as 1-indexed training day (paper axis)
|
||||
red = [0.85 0.10 0.10];
|
||||
blue = [0.10 0.30 0.85];
|
||||
|
||||
[Ma, Sa, na] = localCurve(D, 1, days); % anodal / treatment (stim = 1)
|
||||
[Mc, Sc, nc] = localCurve(D, 0, days); % control (stim = 0)
|
||||
localPlot(D, days, xd, 'count', '# successes', ...
|
||||
fullfile(here, 'learning_curve.png'), vname, red, blue);
|
||||
localPlot(D, days, xd, 'rate', 'success rate', ...
|
||||
fullfile(here, 'learning_curve_rate.png'), vname, red, blue);
|
||||
|
||||
% ------------------------------------------------------------------ helpers
|
||||
function localPlot(D, days, xd, metric, ylab, outFile, vname, red, blue)
|
||||
[Ma, Sa, na] = localCurve(D, 1, days, metric); % anodal / treatment
|
||||
[Mc, Sc, nc] = localCurve(D, 0, days, metric); % control
|
||||
fig = figure('Visible', 'off', 'Color', 'w', 'Position', [100 100 560 460]);
|
||||
hold on
|
||||
e1 = errorbar(xd, Ma, Sa, '-o', 'Color', red, 'MarkerFaceColor', red, 'LineWidth', 2);
|
||||
@@ -30,26 +36,30 @@ e2 = errorbar(xd, Mc, Sc, '-o', 'Color', blue, 'MarkerFaceColor', blue, 'LineWid
|
||||
hold off
|
||||
legend([e1 e2], {sprintf('anodal, N = %d', na), sprintf('control, N = %d', nc)}, ...
|
||||
'Location', 'northwest', 'Box', 'off');
|
||||
xlabel('training day');
|
||||
ylabel('# successes');
|
||||
xlabel('training day'); ylabel(ylab);
|
||||
title(vname, 'Interpreter', 'none');
|
||||
set(gca, 'XTick', xd, 'FontName', 'Arial', 'FontSize', 13, 'LineWidth', 1.5, 'Box', 'off');
|
||||
|
||||
outFile = fullfile(here, 'learning_curve.png');
|
||||
xtickangle(90); % vertical x-axis tick labels
|
||||
exportgraphics(fig, outFile, 'Resolution', 150);
|
||||
close(fig);
|
||||
fprintf('%s: wrote learning_curve.png (anodal N=%d, control N=%d)\n', vname, na, nc);
|
||||
fprintf('%s: wrote %s (anodal N=%d, control N=%d)\n', vname, outFile, na, nc);
|
||||
end
|
||||
|
||||
% ------------------------------------------------------------------ helper
|
||||
function [M, S, n] = localCurve(D, stimVal, days)
|
||||
%LOCALCURVE Per-day mean and SEM of successes across the animals in a group.
|
||||
function [M, S, n] = localCurve(D, stimVal, days, metric)
|
||||
%LOCALCURVE Per-day mean and SEM across the animals in a group, for a metric.
|
||||
subs = unique(D.subject(D.stim == stimVal));
|
||||
n = numel(subs);
|
||||
X = nan(numel(days), n);
|
||||
for j = 1:n
|
||||
for i = 1:numel(days)
|
||||
r = D.subject == subs(j) & D.day == days(i);
|
||||
if any(r); X(i, j) = mean(D.success(r)); end
|
||||
if ~any(r); continue; end
|
||||
if strcmp(metric, 'rate')
|
||||
tot = sum(D.total(r));
|
||||
if tot > 0; X(i, j) = sum(D.success(r)) / tot; end
|
||||
else
|
||||
X(i, j) = mean(D.success(r));
|
||||
end
|
||||
end
|
||||
end
|
||||
M = mean(X, 2, 'omitnan');
|
||||
|
||||
Reference in New Issue
Block a user