ollama部署本地大模型｜embeddinggemma-300m广电行业节目内容标签自动生成-平芜编程栈

ollama部署本地大模型｜embeddinggemma-300m广电行业节目内容标签自动生成

1. 引言

在广电行业，每天都会产生海量的节目内容，如何高效地为这些内容打上准确的标签，是内容管理和检索的关键。传统的人工标注方式不仅耗时耗力，而且难以保证一致性。今天，我将介绍如何使用ollama部署embeddinggemma-300m模型，实现节目内容标签的自动生成。

通过本教程，你将学会：

如何快速部署embeddinggemma-300m模型
如何利用该模型生成文本向量表示
如何应用于广电行业的节目内容标签生成

2. EmbeddingGemma-300m简介

2.1 模型概述

EmbeddingGemma是谷歌推出的开源嵌入模型，参数量为3亿，基于Gemma 3架构开发。这个模型专门用于生成文本的向量表示，非常适合搜索与检索任务，包括分类、聚类及语义相似度搜索。

模型特点：

支持100多种口语语言
体积小巧，适合本地部署
在资源有限的环境（如普通电脑）也能运行良好

2.2 广电行业应用价值

在广电行业，我们可以利用EmbeddingGemma：

自动为节目内容生成标签
实现内容相似度匹配
建立智能内容检索系统
优化内容推荐算法

3. 环境准备与部署

3.1 安装ollama

首先，我们需要安装ollama工具：

curl -fsSL https://ollama.com/install.sh | sh

安装完成后，启动ollama服务：

ollama serve

3.2 下载EmbeddingGemma模型

使用ollama下载embeddinggemma-300m模型：

ollama pull embeddinggemma:300m

下载完成后，可以通过以下命令查看已安装的模型：

ollama list

4. 模型使用与测试

4.1 基本使用示例

让我们先测试模型的基本功能。创建一个Python脚本test_embedding.py：

import ollama # 生成文本嵌入 response = ollama.embeddings( model="embeddinggemma:300m", prompt="这是一档关于人工智能的科技访谈节目" ) print(response["embedding"])

运行脚本：

python test_embedding.py

这将输出文本的向量表示，通常是一个768维的浮点数数组。

4.2 广电节目标签生成示例

下面是一个完整的广电节目内容标签生成示例：

import ollama from sklearn.metrics.pairwise import cosine_similarity import numpy as np # 预定义的标签库 tags = { "科技": "科技前沿、人工智能、科技创新、数字技术", "娱乐": "明星访谈、综艺节目、娱乐新闻、影视资讯", "体育": "体育赛事、运动员专访、体育新闻、健身指导", "新闻": "时事新闻、社会热点、国际动态、政策解读" } # 生成标签嵌入 tag_embeddings = {} for tag, examples in tags.items(): response = ollama.embeddings( model="embeddinggemma:300m", prompt=examples ) tag_embeddings[tag] = response["embedding"] # 待分类的节目描述 program_description = "本期节目将探讨ChatGPT在新闻写作中的应用，并采访了多位AI领域专家" # 生成节目描述的嵌入 response = ollama.embeddings( model="embeddinggemma:300m", prompt=program_description ) program_embedding = response["embedding"] # 计算与各标签的相似度 similarities = {} for tag, embedding in tag_embeddings.items(): similarity = cosine_similarity( [program_embedding], [embedding] )[0][0] similarities[tag] = similarity # 输出最匹配的标签 best_tag = max(similarities, key=similarities.get) print(f"最适合的标签是: {best_tag} (相似度: {similarities[best_tag]:.2f})")

运行结果将显示节目内容最适合的标签。

5. 广电行业实际应用

5.1 批量处理节目内容

在实际应用中，我们通常需要批量处理大量节目内容。以下是一个批量处理的示例：

import ollama import pandas as pd # 读取节目数据 df = pd.read_csv("tv_programs.csv") # 假设有title和description列 # 为每个节目生成标签 def generate_tags(description): # 这里简化处理，实际应用中可以使用更复杂的逻辑 response = ollama.embeddings( model="embeddinggemma:300m", prompt=description ) embedding = response["embedding"] # 这里应该有预先计算好的标签嵌入和相似度计算 # 简化为返回固定标签 return "科技" # 实际应用中应根据相似度返回最匹配的标签 df["generated_tags"] = df["description"].apply(generate_tags) df.to_csv("tv_programs_with_tags.csv", index=False)