Hide keyboard shortcuts

Hot-keys on this page

r m x p   toggle line displays

j k   next/prev highlighted chunk

0   (zero) top of page

1   (one) first highlighted chunk

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

183

184

185

186

187

188

189

190

191

192

193

194

195

196

197

198

199

200

201

202

203

204

205

206

207

208

209

210

211

212

213

214

215

216

217

218

219

220

221

222

223

224

225

226

227

228

229

230

231

232

233

234

235

236

237

238

239

240

241

242

243

244

245

246

247

248

249

250

251

252

253

254

255

256

257

258

259

260

261

262

263

264

265

266

267

268

269

270

271

272

273

274

275

276

277

278

279

280

281

282

283

284

285

286

287

288

289

290

291

292

293

294

295

296

297

298

299

300

301

302

303

304

305

306

307

308

309

310

311

312

313

314

315

316

317

318

319

320

321

322

323

324

325

326

327

328

329

330

331

332

333

334

335

336

337

338

339

340

341

342

343

344

345

346

347

348

349

350

351

352

353

354

355

356

357

358

359

360

361

362

363

364

365

366

367

368

369

370

371

372

373

374

375

376

377

378

379

380

381

382

383

384

385

386

387

388

389

390

391

392

393

394

395

396

397

398

399

400

401

402

403

404

405

406

407

408

409

410

411

412

413

414

415

416

417

418

419

420

421

422

423

424

425

426

427

428

429

430

431

432

433

434

435

436

437

438

439

440

441

442

443

444

445

446

447

448

449

450

451

452

453

454

455

456

457

458

459

460

461

462

463

464

465

466

467

468

469

470

471

472

473

474

475

476

477

478

479

480

481

482

483

484

485

486

487

488

489

490

491

492

493

494

495

496

497

498

499

500

501

502

503

504

505

506

507

508

509

510

511

512

513

514

515

516

517

518

519

520

521

522

523

524

525

526

527

528

529

530

531

532

533

534

535

536

537

538

539

540

541

542

543

544

545

546

547

548

549

550

551

552

553

554

555

556

557

558

559

560

561

562

563

564

565

566

567

568

569

570

571

572

573

574

575

576

577

578

579

580

581

582

583

584

585

586

587

588

589

590

591

592

593

594

595

596

597

598

599

600

601

602

603

604

605

606

607

608

609

610

611

612

613

614

615

616

617

#!/usr/bin/env python3 

# -*- coding: utf-8 -*- 

 

""" 

Postprocessing Text Data, Saving Matrices, Corpora and LDA Models 

***************************************************************** 

 

Functions of this module are for **postprocessing purpose**. You can save \ 

`document-term matrices <https://en.wikipedia.org/wiki/Document-term_matrix>`_, \ 

`tokenized corpora <https://en.wikipedia.org/wiki/Tokenization_(lexical_analysis)>`_ \ 

and `LDA models <https://en.wikipedia.org/wiki/Latent_Dirichlet_allocation>`_, \ 

access topics, topic probabilites for documents, and word probabilities \ 

for each topic. All matrix variants provided in :func:`preprocessing.create_document_term_matrix()`_ \ 

are supported, as well as `lda <https://pypi.python.org/pypi/lda>`_, `Gensim <https://radimrehurek.com/gensim/>`_ \ 

and `MALLET <http://mallet.cs.umass.edu/topics.php>`_ models or output, respectively. \ 

Recurrent variable names are based on the following conventions: 

 

* ``topics`` means a pandas DataFrame containing the top words for each \ 

topic and any Dirichlet parameters. 

* ``document_topics`` means a pandas DataFrame containing topic proportions per \ 

document, at the end of the iterations. 

* ``word_weights`` means unnormalized weights for every topic and word type. 

* ``keys`` means the top *n* tokens of a topic. 

 

Contents 

******** 

* :func:`doc2bow()` 

* :func:`save_document_term_matrix()` writes a document-term matrix to a `CSV <https://en.wikipedia.org/wiki/Comma-separated_values>`_ 

file or to a `Matrix Market <http://math.nist.gov/MatrixMarket/formats.html#MMformat>`_ file, respectively. 

* :func:`save_model()` saves a LDA model (except MALLET models, which will be saved \ 

by specifying a parameter of :func:`mallet.create_mallet_model()`). 

* :func:`save_tokenized_corpus()` writes tokens of a tokenized corpus to plain text \ 

files per document. 

* :func:`show_document_topics()` shows topic probabilities for each document. 

* :func:`show_topics()` shows topics generated by a LDA model. 

* :func:`show_word_weights()` shows word probabilities for each topic. 

""" 

import itertools 

import operator 

import os 

import numpy as np 

import pandas as pd 

import pickle 

import logging 

 

 

log = logging.getLogger(__name__) 

log.addHandler(logging.NullHandler()) 

logging.basicConfig(level=logging.WARNING, 

format='%(levelname)s %(name)s: %(message)s') 

 

 

def doc2bow(document_term_matrix): 

"""Creates a `doc2bow` pandas Series for Gensim. 

 

With this function you can create a `doc2bow` pandas Series as input for Gensim, e.g. \ 

to instantiate the :class:`gensim.models.LdaModel` class or get topic distributions \ 

with :func:`gensim.models.LdaModel.get_document_topics()`. 

 

Args: 

document_term_matrix (pandas.DataFrame): A document-term matrix **designed 

for large corpora**. 

 

Returns: 

List of lists containing tuples. 

 

Todo: 

* Improve efficiency. 

 

Example: 

>>> from dariah_topics import preprocessing 

>>> tokenized_corpus = [['this', 'is', 'document', 'one'], ['this', 'is', 'document', 'two']] 

>>> document_labels = ['document_one', 'document_two'] 

>>> document_term_matrix, _, _ = preprocessing.create_document_term_matrix(tokenized_corpus, document_labels, True) 

>>> isinstance(doc2bow(document_term_matrix), pd.Series) 

True 

""" 

doc2bow = pd.Series() 

for n, document in enumerate(document_term_matrix.index.groupby(document_term_matrix.index.get_level_values('document_id'))): 

doc2bow[str(n)] = [(token, freq) for token, freq in zip(document_term_matrix.loc[document].index, document_term_matrix.loc[document][0])] 

return doc2bow 

 

 

def save_document_term_matrix(document_term_matrix, path, document_ids=None, type_ids=None, matrix_market=False): 

"""Saves document-term matrix. 

 

Writes a ``document_term_matrix`` and, in case of a large corpus matrix, \ 

``document_ids`` and ``type_ids``, which have to be specified, to comma-separated \ 

values (CSV) files. Furthermore, if ``document_term_matrix`` is designed for \ 

large corpora and ``matrix_market`` is True, the matrix will be saved in the \ 

`Matrix Market format <http://math.nist.gov/MatrixMarket/formats.html#MMformat>`_ (`.mm`). \ 

Libraries like `scipy <https://www.scipy.org>`_ and `gensim <https://radimrehurek.com/gensim/>`_ \ 

are able to read and process the Matrix Market format. 

Use the function :func:`preprocessing.create_document_term_matrix()` to create a 

document-term matrix. 

 

Args: 

document_term_matrix (pandas.DataFrame): Document-term matrix with rows 

corresponding to ``document_labels`` and columns corresponding to types 

(unique tokens in the corpus). The single values of the matrix are 

type frequencies. Will be saved as ``document_term_matrix.csv`` or 

``document_term_matrix.mm``, respectively. 

path (str): Path to the output directory. 

document_ids (dict, optional): Dictionary containing ``document_labels`` as 

keys and an unique identifier as value. Only required, if 

``document_term_matrix`` is designed for large corpora. Will be saved 

as ``document_ids.csv``. Defaults to None. 

type_ids (dict, optional): Dictionary containing types as keys and an 

unique identifier as value. Only required, if ``document_term_matrix`` 

is designed for large corpora. Will be saved as ``type_ids.csv``. Defaults 

to None. 

matrix_market (bool, optional): If True, matrix will be saved in Matrix 

Market format. Only for the large corpus variant of ``document_term_matrix`` 

available. Defaults to False. 

 

Returns: 

None. 

 

Example: 

>>> from dariah_topics import preprocessing 

>>> import os 

>>> path = 'tmp' 

>>> tokenized_corpus = [['this', 'is', 'document', 'one'], ['this', 'is', 'document', 'two']] 

>>> document_labels = ['document_one', 'document_two'] 

>>> document_term_matrix = preprocessing.create_document_term_matrix(tokenized_corpus, document_labels) 

>>> save_document_term_matrix(document_term_matrix=document_term_matrix, path=path) 

>>> preprocessing.read_document_term_matrix(os.path.join(path, 'document_term_matrix.csv')) #doctest +NORMALIZE_WHITESPACE 

this is document two one 

document_one 1.0 1.0 1.0 0.0 1.0 

document_two 1.0 1.0 1.0 1.0 0.0 

>>> document_term_matrix, document_ids, type_ids = preprocessing.create_document_term_matrix(tokenized_corpus, document_labels, True) 

>>> save_document_term_matrix(document_term_matrix, path, document_ids, type_ids) 

>>> isinstance(preprocessing.read_document_term_matrix(os.path.join(path, 'document_term_matrix.csv')), pd.DataFrame) 

True 

""" 

if not os.path.exists(path): 

log.info("Creating directory {} ...".format(path)) 

os.makedirs(path) 

if not matrix_market: 

log.info("Saving document_term_matrix.csv to {} ...".format(path)) 

document_term_matrix.to_csv(os.path.join(path, 'document_term_matrix.csv')) 

if isinstance(document_term_matrix.index, pd.MultiIndex) and not matrix_market: 

if document_ids and type_ids is not None: 

log.info("Saving document_ids.csv to {} ...".format(path)) 

pd.Series(document_ids).to_csv(os.path.join(path, 'document_ids.csv')) 

log.info("Saving type_ids.csv to {} ...".format(path)) 

pd.Series(type_ids).to_csv(os.path.join(path, 'type_ids.csv')) 

else: 

raise ValueError("You have to pass document_ids and type_ids as parameters.") 

elif isinstance(document_term_matrix.index, pd.MultiIndex) and matrix_market: 

_save_matrix_market(document_term_matrix, path) 

return None 

 

 

def save_model(model, filepath): 

"""Saves a LDA model. 

 

With this function you can save a LDA model using :module:`pickle`. If you want \ 

to save MALLET models, you have to specify a parameter of the function :func:`mallet.create_mallet_model()`. 

 

Args: 

model: Fitted LDA model produced by `Gensim <https://radimrehurek.com/gensim/>`_ 

or `lda <https://pypi.python.org/pypi/lda>`_. 

filepath (str): Path to LDA model, e.g. ``/home/models/model.pickle``. 

 

Returns: 

None. 

 

Example: 

>>> from lda import LDA 

>>> from gensim.models import LdaModel 

>>> from dariah_topics import preprocessing 

>>> save_model(LDA, 'model.pickle') 

>>> preprocessing.read_model('model.pickle') == LDA 

True 

>>> save_model(LdaModel, 'model.pickle') 

>>> preprocessing.read_model('model.pickle') == LdaModel 

True 

""" 

with open(filepath, 'wb') as file: 

pickle.dump(model, file, protocol=pickle.HIGHEST_PROTOCOL) 

return None 

 

 

def save_tokenized_corpus(tokenized_corpus, document_labels, path): 

"""Writes a tokenized corpus to text files. 

 

With this function you can write tokens of a `tokenized_corpus` to plain text \ 

files per document to ``path``. Every file will be named after its ``document_label``. \ 

Depending on the used tokenizer, ``tokenized_corpus`` does normally not contain \ 

any punctuations or one-letter words. 

Use the function :func:`preprocessing.tokenize()` to tokenize a corpus. 

 

Args: 

tokenized_corpus (list): Tokenized corpus containing one or more 

iterables containing tokens. 

document_labels (list): Name of each `tokenized_document` in `tokenized_corpus`. 

path (str): Path to the output directory. 

 

Returns: 

None 

 

Example: 

>>> tokenized_corpus = [['this', 'is', 'a', 'tokenized', 'document']] 

>>> document_labels = ['document_label'] 

>>> path = 'tmp' 

>>> save_tokenized_corpus(tokenized_corpus, document_labels, path) 

>>> with open(os.path.join(path, 'document_label.txt'), 'r', encoding='utf-8') as file: 

... file.read() 

'this\\nis\\na\\ntokenized\\ndocument' 

""" 

log.info("Saving tokenized corpus to {} ...".format(path)) 

if not os.path.exists(path): 

log.info("Creating directory {} ...".format(path)) 

os.makedirs(path) 

 

for tokenized_document, document_label in zip(tokenized_corpus, document_labels): 

log.debug("Current file: {}".format(document_label)) 

with open(os.path.join(path, '{}.txt'.format(document_label)), 'w', encoding='utf-8') as file: 

file.write('\n'.join(tokenized_document)) 

return None 

 

 

def show_document_topics(topics=None, model=None, document_labels=None, doc_topics_file=None, doc2bow=None, num_keys=3): 

"""Shows topic distribution for each document. 

 

With this function you can show the topic distributions for all documents in a pandas DataFrame. \ 

For each topic, the top ``num_keys`` keys will be considered. If you have a 

* `lda <https://pypi.python.org/pypi/lda>`_ model, you have to pass the model \ 

as ``model`` and the document-term matrix vocabulary as ``vocabulary``. 

* `Gensim <https://radimrehurek.com/gensim/>`_ model, you have to pass only the model \ 

as ``model``. 

* `MALLET <http://mallet.cs.umass.edu/topics.php>`_ based workflow, you have to\ 

pass only the ``doc_topics_file``. 

 

Args: 

topics (pandas.DataFrame, optional): Only for lda models. A pandas DataFrame 

containing all topics. 

model (optional): lda or Gensim model. 

document_labels (list, optional): An list of all document labels. 

doc_topics_file (str, optional): Only for MALLET. Path to the doc-topics file. 

doc2bow (list, optional): A list of lists containing tuples of ``type_id`` and 

frequency. 

num_keys (int, optional): Number of top keys for each topic. 

 

Returns: 

A pandas DataFrame with rows corresponding to topics and columns corresponding 

to keys. 

 

Example: 

""" 

from lda.lda import LDA 

from gensim.models import LdaModel, LdaMulticore 

 

index = [' '.join(keys[:num_keys]) for keys in topics.values] 

if isinstance(model, LDA): 

return _show_lda_document_topics(model, document_labels, index) 

elif isinstance(model, LdaModel) or isinstance(model, LdaMulticore): 

return _show_gensim_document_topics(doc2bow, model, document_labels, index) 

elif doc_topics_file is not None: 

return _show_mallet_document_topics(doc_topics_file, index) 

 

 

def show_topics(model=None, vocabulary=None, topic_keys_file=None, num_keys=10): 

"""Shows topics of LDA model. 

 

With this function you can show all topics of a LDA model in a pandas DataFrame. \ 

For each topic, the top ``num_keys`` keys will be considered. If you have a 

* `lda <https://pypi.python.org/pypi/lda>`_ model, you have to pass the model \ 

as ``model`` and the document-term matrix vocabulary as ``vocabulary``. 

* `Gensim <https://radimrehurek.com/gensim/>`_ model, you have to pass only the model \ 

as ``model``. 

* `MALLET <http://mallet.cs.umass.edu/topics.php>`_ based workflow, you have to\ 

pass only the ``topic_keys_file``. 

 

Args: 

model (optional): lda or Gensim model. 

vocabulary (list, optional): Only for lda. The vocabulary of the  

document-term matrix. 

topic_keys_file (str): Only for MALLET. Path to the topic keys file. 

num_keys (int, optional): Number of top keys for each topic.  

 

Returns: 

A pandas DataFrame with rows corresponding to topics and columns corresponding 

to keys. 

 

Example: 

""" 

from lda.lda import LDA 

from gensim.models import LdaModel, LdaMulticore 

 

if isinstance(model, LDA): 

return _show_lda_topics(model, vocabulary, num_keys) 

elif isinstance(model, LdaModel) or isinstance(model, LdaMulticore): 

return _show_gensim_topics(model, num_keys) 

elif topic_keys_file is not None: 

return _show_mallet_topics(topic_keys_file) 

 

 

def show_word_weights(word_weights_file, num_tokens): 

"""Read Mallet word_weigths file 

 

Description: 

Reads Mallet word_weigths into pandas DataFrame. 

 

Args: 

word_weigts_file: Word_weights_file created with Mallet 

 

Returns: Pandas DataFrame 

 

Todo: 

* Adapt for ``lda`` and ``gensim`` output. 

 

Example: 

>>> import tempfile 

>>> with tempfile.NamedTemporaryFile(suffix='.txt') as tmpfile: 

... tmpfile.write(b'0\\tthis\\t0.5\\n0\\tis\\t0.4\\n0\\ta\\t0.3\\n0\\tdocument\\t0.2') and True 

... tmpfile.flush() 

... show_word_weights(tmpfile.name, 2) #doctest: +NORMALIZE_WHITESPACE 

True 

document token weight 

0 0 this 0.5 

1 0 is 0.4 

 

""" 

word_weights = pd.read_table(word_weights_file, header=None, sep='\t', names=['document', 'token', 'weight']) 

return word_weights.sort_values('weight', ascending=False)[:num_tokens] 

 

 

def _grouper(n, iterable, fillvalue=None): 

"""Collects data into fixed-length chunks or blocks. 

 

This private function is wrapped in :func:`_show_mallet_document_topics()`. 

 

Args: 

n (int): Length of chunks or blocks 

iterable (object): Iterable object 

fillvalue (boolean): If iterable can not be devided into evenly-sized chunks fill chunks with value. 

 

Returns: n-sized chunks 

 

""" 

args = [iter(iterable)] * n 

return itertools.zip_longest(*args, fillvalue=fillvalue) 

 

 

def _show_gensim_document_topics(doc2bow, model, document_labels, index): 

"""Creates a document-topic-matrix. 

 

Description: 

With this function you can create a doc-topic-maxtrix for gensim  

output.  

 

Args: 

corpus (mmCorpus): Gensim corpus. 

model: Gensim LDA model 

doc_labels (list): List of document labels. 

 

Returns:  

Doc_topic-matrix as DataFrame 

 

Example: 

>>> from gensim.models import LdaModel 

>>> from gensim.corpora import Dictionary 

>>> document_labels = ['document_one', 'document_two'] 

>>> tokenized_corpus = [['this', 'is', 'the', 'first', 'document'], ['this', 'is', 'the', 'second', 'document']] 

>>> id2word = Dictionary(tokenized_corpus) 

>>> corpus = [id2word.doc2bow(document) for document in tokenized_corpus] 

>>> model = LdaModel(corpus=corpus, id2word=id2word, iterations=1, passes=1, num_topics=2) 

>>> topics = _show_gensim_topics(model, 5) 

>>> index = [' '.join(keys[:2]) for keys in topics.values] 

>>> isinstance(_show_gensim_document_topics(corpus, model, document_labels, index), pd.DataFrame) 

True 

""" 

num_topics = model.num_topics 

num_documents = len(document_labels) 

document_topics = np.zeros((num_topics, num_documents)) 

 

for n, document in enumerate(doc2bow): 

for distribution in model.get_document_topics(document): 

document_topics[distribution[0]][n] = distribution[1] 

return pd.DataFrame(document_topics, index=index, columns=document_labels) 

 

 

def _show_gensim_topics(model, num_keys=10): 

"""Converts gensim output to DataFrame. 

 

Description: 

With this function you can convert gensim output (usually a list of 

tuples) to a DataFrame, a more convenient datastructure. 

 

Args: 

model: Gensim LDA model. 

num_keys (int): Number of top keywords for topic. 

 

Returns: 

DataFrame. 

 

ToDo: 

 

Example: 

>>> from gensim.models import LdaModel 

>>> from gensim.corpora import Dictionary 

>>> tokenized_corpus = [['this', 'is', 'the', 'first', 'document'], ['this', 'is', 'the', 'second', 'document']] 

>>> id2word = Dictionary(tokenized_corpus) 

>>> corpus = [id2word.doc2bow(document) for document in tokenized_corpus] 

>>> model = LdaModel(corpus=corpus, id2word=id2word, iterations=1, passes=1, num_topics=2) 

>>> isinstance(_show_gensim_topics(model, 5), pd.DataFrame) 

True 

""" 

log.info("Accessing topics from Gensim model ...") 

topics = [] 

for n, topic in model.show_topics(formatted=False, num_words=num_keys): 

topics.append([key[0] for key in topic]) 

index = ['Topic {}'.format(n) for n in range(len(topics))] 

columns = ['Key {}'.format(n) for n in range(num_keys)] 

return pd.DataFrame(topics, index=index, columns=columns) 

 

 

def _show_lda_document_topics(model, document_labels, index): 

"""Creates a doc_topic_matrix for lda output. 

 

Description: 

With this function you can convert lda output to a DataFrame,  

a more convenient datastructure. 

Use 'lda2DataFrame()' to get topics. 

 

Note: 

 

Args: 

model: Gensim LDA model. 

topics: DataFrame. 

doc_labels (list[str]): List of doc labels as string. 

 

Returns: 

DataFrame 

 

Example: 

>>> import lda 

>>> from dariah_topics import preprocessing 

>>> tokenized_corpus = [['this', 'is', 'the', 'first', 'document'], ['this', 'is', 'the', 'second', 'document']] 

>>> document_labels = ['document_one', 'document_two'] 

>>> document_term_matrix = preprocessing.create_document_term_matrix(tokenized_corpus, document_labels) 

>>> vocabulary = document_term_matrix.columns 

>>> model = lda.LDA(n_topics=2, n_iter=1) 

>>> model = model.fit(document_term_matrix.as_matrix().astype(int)) 

>>> topics = _show_lda_topics(model, vocabulary, num_keys=5) 

>>> index = [' '.join(keys[:3]) for keys in topics.values] 

>>> isinstance(_show_lda_document_topics(model, document_labels, index), pd.DataFrame) 

True 

""" 

return pd.DataFrame(model.doc_topic_, index=document_labels, columns=index).T 

 

 

def _show_lda_topics(model, vocabulary, num_keys): 

"""Converts lda output to a DataFrame 

 

Description: 

With this function you can convert lda output to a DataFrame,  

a more convenient datastructure. 

 

Note: 

 

Args: 

model: LDA model. 

vocab (list[str]): List of strings containing corpus vocabulary.  

num_keys (int): Number of top keywords for topic 

 

Returns: 

DataFrame 

 

Example: 

>>> import lda 

>>> from dariah_topics import preprocessing 

>>> tokenized_corpus = [['this', 'is', 'the', 'first', 'document'], ['this', 'is', 'the', 'second', 'document']] 

>>> document_labels = ['document_one', 'document_two'] 

>>> document_term_matrix = preprocessing.create_document_term_matrix(tokenized_corpus, document_labels) 

>>> vocabulary = document_term_matrix.columns 

>>> model = lda.LDA(n_topics=2, n_iter=1) 

>>> model = model.fit(document_term_matrix.as_matrix().astype(int)) 

>>> isinstance(_show_lda_topics(model, vocabulary, num_keys=5), pd.DataFrame) 

True 

""" 

log.info("Accessing topics from lda model ...") 

topics = [] 

topic_word = model.topic_word_ 

for i, topic_distribution in enumerate(topic_word): 

topics.append(np.array(vocabulary)[np.argsort(topic_distribution)][:-num_keys-1:-1]) 

index = ['Topic {}'.format(n) for n in range(len(topics))] 

columns = ['Key {}'.format(n) for n in range(num_keys)] 

return pd.DataFrame(topics, index=index, columns=columns) 

 

 

def _show_mallet_document_topics(doc_topics_file, index): 

"""Shows document-topic-mapping. 

Args: 

outfolder (str): Folder for MALLET output. 

doc_topics (str): Name of MALLET's doc_topic file. Defaults to 'doc_topics.txt'. 

topic_keys (str): Name of MALLET's topic_keys file. Defaults to 'topic_keys.txt'. 

 

ToDo: Prettify docnames 

 

Example: 

>>> import tempfile 

>>> index = ['first topic', 'second topic'] 

>>> with tempfile.NamedTemporaryFile(suffix='.txt') as tmpfile: 

... tmpfile.write(b'0\\tdocument_one.txt\\t0.1\\t0.2\\n1\\tdocument_two.txt\\t0.4\\t0.5') and True 

... tmpfile.flush() 

... _show_mallet_document_topics(tmpfile.name, index) #doctest: +NORMALIZE_WHITESPACE 

True 

document_one document_two 

first topic 0.1 0.4 

second topic 0.2 0.5 

""" 

document_topics_triples = [] 

document_labels = [] 

topics = [] 

with open(doc_topics_file, 'r', encoding='utf-8') as file: 

for line in file: 

l = line.lstrip() 

if l.startswith('#'): 

lines = file.readlines() 

for line in lines: 

documet_number, document_label, *values = line.rstrip().split('\t') 

document_labels.append(os.path.splitext(os.path.basename(document_label))[0]) 

for topic, share in _grouper(2, values): 

triple = (document_label, int(topic), float(share)) 

topics.append(int(topic)) 

document_topics_triples.append(triple) 

else: 

easy_file_format = True 

break 

if easy_file_format: 

document_topics = pd.read_table(doc_topics_file, sep='\t', header=None) 

document_topics.index = [os.path.splitext(os.path.basename(document_label))[0] for document_label in document_topics[1]] 

document_topics = document_topics.drop([0, 1], axis=1) 

document_topics.columns = index 

return document_topics.T 

else: 

document_topics_triples = sorted(document_topics_triples, key=operator.itemgetter(0, 1)) 

document_labels = sorted(document_labels) 

num_documents = len(document_labels) 

num_topics = len(topics) 

document_topics = np.zeros((num_documents, num_topics)) 

for triple in document_topics_triples: 

document_label, topic, share = triple 

index_num = document_labels.index(document_label) 

document_topics[index_num, topic] = share 

return pd.DataFrame(document_topics, index=index, columns=columns.T) 

 

 

def _show_mallet_topics(path_to_topic_keys_file): 

"""Show topic-key-mapping. 

 

Args: 

outfolder (str): Folder for Mallet output, 

topicsKeyFile (str): Name of Mallets' topic_key file, default "topic_keys" 

 

#topic-model-mallet 

Note: FBased on DARIAH-Tutorial -> https://de.dariah.eu/tatom/topic_model_mallet.html 

 

ToDo: Prettify index 

 

Example:  

>>> import tempfile 

>>> with tempfile.NamedTemporaryFile(suffix='.txt') as tmpfile: 

... tmpfile.write(b'0\\t0.5\\tthis is the first document\\n1\\t0.5\\tthis is the second document') and True 

... tmpfile.flush() 

... _show_mallet_topics(tmpfile.name) 

True 

Key 0 Key 1 Key 2 Key 3 Key 4 

Topic 0 this is the first document 

Topic 1 this is the second document 

""" 

log.info("Accessing topics from MALLET model ...") 

topics = [] 

with open(path_to_topic_keys_file, 'r', encoding='utf-8') as file: 

for line in file.readlines(): 

_, _, keys = line.split('\t') 

keys = keys.rstrip().split(' ') 

topics.append(keys) 

index = ['Topic {}'.format(n) for n in range(len(topics))] 

columns = ['Key {}'.format(n) for n in range(len(topics[0]))] 

return pd.DataFrame(topics, index=index, columns=columns) 

 

 

def _save_matrix_market(document_term_matrix, path): 

""" 

Writes a `document_term_matrix` designed for large corpora to `Matrix Market <http://math.nist.gov/MatrixMarket/formats.html#MMformat>`_ file (`.mm`). Libraries like `scipy <https://www.scipy.org>`_ 

and `gensim <https://radimrehurek.com/gensim/>`_ are able to read and process 

the Matrix Market format. This private function is wrapped in `save_document_term_matrix()`. 

 

**Use the function `preprocessing.create_document_term_matrix()` to create a 

document-term matrix.** 

 

Args: 

document_term_matrix (pandas.DataFrame): Document-term matrix with only 

one column corresponding to type frequencies and a pandas MultiIndex 

with `document_ids` for level 0 and `type_ids` for level 1. Will be 

saved as `document_term_matrix.mm`. 

path (str): Path to the output directory. 

 

Returns: 

None. 

 

Example: 

""" 

num_docs = document_term_matrix.index.get_level_values('document_id').max() 

num_types = document_term_matrix.index.get_level_values('type_id').max() 

sum_counts = document_term_matrix[0].sum() 

header = "{} {} {}\n".format(num_docs, num_types, sum_counts) 

 

with open(os.path.join(path, 'document_term_matrix.mm'), 'w', encoding='utf-8') as file: 

file.write("%%MatrixMarket matrix coordinate real general\n") 

file.write(header) 

document_term_matrix.to_csv(file, sep=' ', header=None) 

return None