Western diets are high in fat and sucrose and can influence behavior and gut microbiota. There is growing evidence that altering the microbiome can influence the brain and behavior. This study was designed to determine whether diet-induced changes in the gut microbiota could contribute to alterations in anxiety, memory or cognitive flexibility. Two-month-old, male C57BL/6 mice were randomly assigned high-fat (42% fat, 43% carbohydrate (CHO), high-sucrose (12% fat, 70% CHO (primarily sucrose) or normal chow (13% kcal fat, 62% CHO) diets. Fecal microbiome analysis, step-down latency, novel object and novel location tasks were performed prior to and 2weeks after diet change. Water maze testing for long- and short-term memory and cognitive flexibility was conducted during weeks 5-6 post-diet change. Some similarities in alterations in the microbiome were seen in both the high-fat and high-sucrose diets (e.g., increased Clostridiales), as compared to the normal diet, but the percentage decreases in Bacteroidales were greater in the high-sucrose diet mice. Lactobacillales was only significantly increased in the high-sucrose diet group and Erysipelotrichales was only significantly affected by the high-fat diet. The high-sucrose diet group was significantly impaired in early development of a spatial bias for long-term memory, short-term memory and reversal training, compared to mice on normal diet. An increased focus on the former platform position was seen in both high-sucrose and high-fat groups during the reversal probe trials. There was no significant effect of diet on step-down, exploration or novel recognitions. Higher percentages of Clostridiales and lower expression of Bacteroidales in high-energy diets were related to the poorer cognitive flexibility in the reversal trials. These results suggest that changes in the microbiome may contribute to cognitive changes associated with eating a Western diet.